experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #4143 from pmenglund/image_retraining_error_message

Browse Source 
improve error message when a category is empty
This commit is contained in:
Vincent Vanhoucke 2016-09-03 18:53:42 -07:00 committed by GitHub
commit 68a74ea985
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/examples/image_retraining/retrain.py
View File

@ -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']