diff options
author | Dan Mané <danmane@google.com> | 2016-09-09 16:07:46 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-09 17:19:12 -0700 |
commit | 54a7178849a7603de919e6cdc4f404b470f87d14 (patch) | |
tree | b0fe52e6330545f45541504257ab2c9da626be2f /tensorflow/examples/image_retraining | |
parent | f51e1964b57ecce5f3b38d4156cf61e80e5c4181 (diff) |
Merge changes from github.
Change: 132733397
Diffstat (limited to 'tensorflow/examples/image_retraining')
-rw-r--r-- | tensorflow/examples/image_retraining/retrain.py | 5 |
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) |