Skip to content

GitLab

Commit c59cf48d authored by Yukun Zhu's avatar Yukun Zhu Committed by aquariusjay
Browse files

Fix typo in train.py (#7805) 

Thanks to issue raised at https://github.com/tensorflow/models/issues/7799
parent 61aee86e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deeplab/train.py
View file @ c59cf48d
...@@ -342,7 +342,7 @@ def main(unused_argv): ...@@ -342,7 +342,7 @@ def main(unused_argv):
first_clone_label = graph.get_tensor_by_name( first_clone_label = graph.get_tensor_by_name(
('%s/%s:0' % (first_clone_scope, common.LABEL)).strip('/')) ('%s/%s:0' % (first_clone_scope, common.LABEL)).strip('/'))
# Scale up summary image pixel values for better visualization. # Scale up summary image pixel values for better visualization.
pixel_scaling = max(1, 255 // dataset.num_classes) pixel_scaling = max(1, 255 // dataset.num_of_classes)
summary_label = tf.cast(first_clone_label * pixel_scaling, tf.uint8) summary_label = tf.cast(first_clone_label * pixel_scaling, tf.uint8)
summaries.add( summaries.add(
tf.summary.image('samples/%s' % common.LABEL, summary_label)) tf.summary.image('samples/%s' % common.LABEL, summary_label))
......
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