Merge pull request tensorflow#5021 from sumsuddin/constant_ext_len_fix

YknZhu · web-flow · commit fb6bc29b236f · 2018-08-07T09:59:00.000-07:00
fix for constant extension length
diff --git a/research/deeplab/datasets/remove_gt_colormap.py b/research/deeplab/datasets/remove_gt_colormap.py
@@ -72,7 +72,7 @@ def main(unused_argv):
                                        '*.' + FLAGS.segmentation_format))
   for annotation in annotations:
     raw_annotation = _remove_colormap(annotation)
-    filename = os.path.basename(annotation)[:-4]
+    filename = os.path.splitext(os.path.basename(annotation))[0]
     _save_annotation(raw_annotation,
                      os.path.join(
                          FLAGS.output_dir,
