fix for constant extension length

f1d3764e · Shojib · 308179f3 · f1d3764e
Commit f1d3764e authored Aug 07, 2018 by Shojib
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/deeplab/datasets/remove_gt_colormap.py research/deeplab/datasets/remove_gt_colormap.py +1 -1

No files found.
--- 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,