diff --git a/tensorflow/examples/image_retraining/retrain.py b/tensorflow/examples/image_retraining/retrain.py index acd3005c483..3becd7f5721 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']