aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/image_retraining
diff options
context:
space:
mode:
authorGravatar Dan Mané <danmane@google.com>2016-09-09 16:07:46 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-09-09 17:19:12 -0700
commit54a7178849a7603de919e6cdc4f404b470f87d14 (patch)
treeb0fe52e6330545f45541504257ab2c9da626be2f /tensorflow/examples/image_retraining
parentf51e1964b57ecce5f3b38d4156cf61e80e5c4181 (diff)
Merge changes from github.
Change: 132733397
Diffstat (limited to 'tensorflow/examples/image_retraining')
-rw-r--r--tensorflow/examples/image_retraining/retrain.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/examples/image_retraining/retrain.py b/tensorflow/examples/image_retraining/retrain.py
index acd3005c48..f226de0ce3 100644
--- a/tensorflow/examples/image_retraining/retrain.py
+++ b/tensorflow/examples/image_retraining/retrain.py
@@ -268,7 +268,8 @@ def get_image_path(image_lists, label_name, index, image_dir, category):
tf.logging.fatal('Category does not exist %s.', category)
category_list = label_lists[category]
if not category_list:
- tf.logging.fatal('Category has no images - %s.', category)
+ tf.logging.fatal('Label %s has no images in the category %s.',
+ label_name, category)
mod_index = index % len(category_list)
base_name = category_list[mod_index]
sub_dir = label_lists['dir']
@@ -704,7 +705,7 @@ def variable_summaries(var, name):
tf.scalar_summary('mean/' + name, mean)
with tf.name_scope('stddev'):
stddev = tf.sqrt(tf.reduce_mean(tf.square(var - mean)))
- tf.scalar_summary('sttdev/' + name, stddev)
+ tf.scalar_summary('stddev/' + name, stddev)
tf.scalar_summary('max/' + name, tf.reduce_max(var))
tf.scalar_summary('min/' + name, tf.reduce_min(var))
tf.histogram_summary(name, var)