fix for constant extension length

sumsuddin · sumsuddin · commit f1d3764effa0 · 2018-08-07T13:48:43.000+06:00
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,
