Skip to content

GitLab

Commit 5281c9a0 authored by Yubin Ruan's avatar Yubin Ruan
Browse files

vis.py: use basename of image when saving raw annotations to prevent file-not-found error

parent 1d38a225
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deeplab/vis.py
View file @ 5281c9a0
...@@ -173,7 +173,7 @@ def _process_batch(sess, original_images, semantic_predictions, image_names, ...@@ -173,7 +173,7 @@ def _process_batch(sess, original_images, semantic_predictions, image_names,
colormap_type=FLAGS.colormap_type) colormap_type=FLAGS.colormap_type)
if FLAGS.also_save_raw_predictions: if FLAGS.also_save_raw_predictions:
image_filename = image_names[i] image_filename = os.path.basename(image_names[i])
if train_id_to_eval_id is not None: if train_id_to_eval_id is not None:
crop_semantic_prediction = _convert_train_id_to_eval_id( crop_semantic_prediction = _convert_train_id_to_eval_id(
......
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