Skip to content

GitLab

Commit 8f6daf5d authored by Billy Lamberta's avatar Billy Lamberta
Browse files

Merge branch 'master' of github.com:tensorflow/models into fix-nb-eager

parents 5a060b4f 46377f88
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deeplab/vis.py
View file @ 8f6daf5d
...@@ -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