2016-12-26 1 views
0

Итак, я пытаюсь переустановить модель классификатора изображений Inception 3 через их код rerain.py из своего github («https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py») (Код действительно длинный, поэтому я решил, что не должен размещать его здесь)Переподготовка изображений Tensorflow с помощью Inception3; где должны храниться новые изображения?

Однако я не могу понять, как правильно хранить изображения, которые я хочу использовать, или в какой директории я должен хранить их. Я использую Windows 7 с Python 3.5

В настоящее время у меня есть изображения в папке с именем Images, которые далее разбиваются на подпапки, названные в соответствии с ярлыками, которые я хочу.

Так как пример, это что-то вроде Images -> 'Cats' -> 'Dogs' -> 'Fish'

где изображения является основной папки и вложенные папки являются виды животных.

Папка Images находится в местоположении C:\tmp\, который является местом, где учебник Mnist автоматически загружал набор данных числа Mnist, когда я запустил mnist tutorial.py. Поэтому я решил, что должен хранить свои изображения в одном месте.

Когда я бегу retrain.py по ссылке выше GitHub:

Я получаю следующую ошибку

Image directory "" not found. File "C:\Users\Amit\AppData\Local\Programs\Python\Python35\Lib\site-packages\tensorflow\models\image\imagenet\tf_retrain.py", line 668, in main class_count = len(image_lists.keys()) AttributeError: 'NoneType' object has no attribute 'keys'

Я просмотрел код и пытались найти значение по умолчанию для Image_dir, который является аргументом для функции def create_image_lists (которая, кажется, там, где сценарий ищет изображения, но не может найти его, или, может быть, я неправильно понимаю код.

EDIT: У меня возникли проблемы с запуском python 3.5 через командную строку, поэтому я не могу использовать argparse.

спасибо.

ответ

Смежные вопросы