提交 c59cf48d 编写于 作者: Y Yukun Zhu 提交者: aquariusjay

Fix typo in train.py (#7805)

Thanks to issue raised at https://github.com/tensorflow/models/issues/7799
上级 61aee86e
......@@ -342,7 +342,7 @@ def main(unused_argv):
first_clone_label = graph.get_tensor_by_name(
('%s/%s:0' % (first_clone_scope, common.LABEL)).strip('/'))
# 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)
summaries.add(
tf.summary.image('samples/%s' % common.LABEL, summary_label))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册