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
......@@ -173,7 +173,7 @@ def _process_batch(sess, original_images, semantic_predictions, image_names,
colormap_type=FLAGS.colormap_type)
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:
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