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

Merge pull request #3766 from walkerlala/myrepo/master

vis.py: use basename of image when saving raw annotations to prevent file-not-found error
parents 365ea208 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