Skip to content

GitLab

Unverified Commit fb6bc29b authored by Yukun Zhu's avatar Yukun Zhu Committed by GitHub
Browse files

Merge pull request #5021 from sumsuddin/constant_ext_len_fix 

fix for constant extension length
parents 5fd32ef6 f1d3764e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deeplab/datasets/remove_gt_colormap.py
View file @ fb6bc29b
...@@ -72,7 +72,7 @@ def main(unused_argv): ...@@ -72,7 +72,7 @@ def main(unused_argv):
'*.' + FLAGS.segmentation_format)) '*.' + FLAGS.segmentation_format))
for annotation in annotations: for annotation in annotations:
raw_annotation = _remove_colormap(annotation) raw_annotation = _remove_colormap(annotation)
filename = os.path.basename(annotation)[:-4] filename = os.path.splitext(os.path.basename(annotation))[0]
_save_annotation(raw_annotation, _save_annotation(raw_annotation,
os.path.join( os.path.join(
FLAGS.output_dir, FLAGS.output_dir,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment