Replace obsolete import contextlib2 with the equivalent stdlib module contextlib

PiperOrigin-RevId: 769153291

Author: oprypin

research/object_detection/dataset_tools/create_coco_tf_record.py

@@ -33,15 +33,15 @@
 from __future__ import division
 from __future__ import print_function
 
+import contextlib
 import hashlib
 import io
 import json
 import logging
 import os
-import contextlib2
+
 import numpy as np
 import PIL.Image
-
 from pycocotools import mask
 import tensorflow.compat.v1 as tf
 
@@ -385,8 +385,9 @@ def _create_tf_record_from_coco_annotations(annotations_file, image_dir,
     remove_non_person_images: Whether to remove any images that do not contain
       at least one "person" annotation.
   """
-  with contextlib2.ExitStack() as tf_record_close_stack, \
-      tf.gfile.GFile(annotations_file, 'r') as fid:
+  with contextlib.ExitStack() as tf_record_close_stack, tf.gfile.GFile(
+      annotations_file, 'r'
+  ) as fid:
     output_tfrecords = tf_record_creation_util.open_sharded_output_tfrecords(
         tf_record_close_stack, output_path, num_shards)
     groundtruth_data = json.load(fid)

research/object_detection/dataset_tools/create_oid_tf_record.py

@@ -36,9 +36,9 @@
 from __future__ import division
 from __future__ import print_function
 
+import contextlib
 import os
 
-import contextlib2
 import pandas as pd
 import tensorflow.compat.v1 as tf
 
@@ -91,7 +91,7 @@ def main(_):
 
   tf.logging.log(tf.logging.INFO, 'Found %d images...', len(all_image_ids))
 
-  with contextlib2.ExitStack() as tf_record_close_stack:
+  with contextlib.ExitStack() as tf_record_close_stack:
     output_tfrecords = tf_record_creation_util.open_sharded_output_tfrecords(
         tf_record_close_stack, FLAGS.output_tf_record_path_prefix,
         FLAGS.num_shards)

research/object_detection/dataset_tools/create_pet_tf_record.py

@@ -26,14 +26,14 @@
         --output_dir=/home/user/pet/output
 """
 
+import contextlib
 import hashlib
 import io
 import logging
 import os
 import random
 import re
 
-import contextlib2
 from lxml import etree
 import numpy as np
 import PIL.Image
@@ -233,7 +233,7 @@ def create_tf_record(output_filename,
     mask_type: 'numerical' or 'png'. 'png' is recommended because it leads to
       smaller file sizes.
   """
-  with contextlib2.ExitStack() as tf_record_close_stack:
+  with contextlib.ExitStack() as tf_record_close_stack:
     output_tfrecords = tf_record_creation_util.open_sharded_output_tfrecords(
         tf_record_close_stack, output_filename, num_shards)
     for idx, example in enumerate(examples):

research/object_detection/dataset_tools/tf_record_creation_util_test.py

@@ -18,8 +18,9 @@
 from __future__ import division
 from __future__ import print_function
 
+import contextlib
 import os
-import contextlib2
+
 import six
 from six.moves import range
 import tensorflow.compat.v1 as tf
@@ -30,7 +31,7 @@
 class OpenOutputTfrecordsTests(tf.test.TestCase):
 
   def test_sharded_tfrecord_writes(self):
-    with contextlib2.ExitStack() as tf_record_close_stack:
+    with contextlib.ExitStack() as tf_record_close_stack:
       output_tfrecords = tf_record_creation_util.open_sharded_output_tfrecords(
           tf_record_close_stack,
           os.path.join(tf.test.get_temp_dir(), 'test.tfrec'), 10)