Make this file TensorFlow 2.0 compatible (#7836)

* Make this file TensorFlow 2.0 compatible

* Update remove_gt_colormap.py

research/deeplab/datasets/remove_gt_colormap.py

@@ -26,15 +26,15 @@ from PIL import Image
 
 import tensorflow as tf
 
-FLAGS = tf.app.flags.FLAGS
+FLAGS = tf.compat.v1.flags.FLAGS
 
-tf.app.flags.DEFINE_string('original_gt_folder',
+tf.compat.v1.flags.DEFINE_string('original_gt_folder',
                                  './VOCdevkit/VOC2012/SegmentationClass',
                                  'Original ground truth annotations.')
 
-tf.app.flags.DEFINE_string('segmentation_format', 'png', 'Segmentation format.')
+tf.compat.v1.flags.DEFINE_string('segmentation_format', 'png', 'Segmentation format.')
 
-tf.app.flags.DEFINE_string('output_dir',
+tf.compat.v1.flags.DEFINE_string('output_dir',
                                  './VOCdevkit/VOC2012/SegmentationClassRaw',
                                  'folder to save modified ground truth annotations.')
 
@@ -59,14 +59,14 @@ def _save_annotation(annotation, filename):
     filename: Output filename.
   """
   pil_image = Image.fromarray(annotation.astype(dtype=np.uint8))
-  with tf.gfile.Open(filename, mode='w') as f:
+  with tf.io.gfile.GFile(filename, mode='w') as f:
     pil_image.save(f, 'PNG')
 
 
 def main(unused_argv):
   # Create the output directory if not exists.
-  if not tf.gfile.IsDirectory(FLAGS.output_dir):
-    tf.gfile.MakeDirs(FLAGS.output_dir)
+  if not tf.io.gfile.isdir(FLAGS.output_dir):
+    tf.io.gfile.makedirs(FLAGS.output_dir)
 
   annotations = glob.glob(os.path.join(FLAGS.original_gt_folder,
                                        '*.' + FLAGS.segmentation_format))
@@ -80,4 +80,4 @@ def main(unused_argv):
 
 
 if __name__ == '__main__':
-  tf.app.run()
+  tf.compat.v1.app.run()