Я не могу сделать комментарий к вашему Вопрос, так как у меня пока нет кредитов. Поэтому позвольте мне дать вам общий ответ.
Файл inception-2015-12-05.tgz
вы упомянули содержит два файла, которые вам требуется:
а) imagenet_comp_graph_label_strings.txt
б) tensorflow_inception_graph.pb
Существует файл лицензии, который вам также не потребуется. Эти два файла позволят вам делать прогнозы на изображения.
Часть, в которой вы упоминали the model was not saved after a reboot, so I had to download it again via the same script
интригует меня. Я никогда не сталкивался с такой проблемой. Попробуйте это сейчас:
- Создайте папку в выбранном вами месте. Скажите
~/Documents
.
- При запуске скрипта python
classify_image.py
используйте флаг --model_dir
, чтобы перенаправить каталог файла модели на ~/Documents
. Это будет по существу загружать и извлекать необходимые файлы в указанное место, и вы можете использовать то же местоположение в --model_dir
флагах с тех пор.
Взгляните на это:
Aruns-MacBook-Pro:imagenet arundas$ python classify_image.py --model_dir ~/Documents/
>> Downloading inception-2015-12-05.tgz 100.0%
Succesfully downloaded inception-2015-12-05.tgz 88931400 bytes.
W tensorflow/core/framework/op_def_util.cc:332] Op BatchNormWithGlobalNormalization is deprecated. It will cease to work in GraphDef version 9. Use tf.nn.batch_normalization().
giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca (score = 0.89233)
indri, indris, Indri indri, Indri brevicaudatus (score = 0.00859)
lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens (score = 0.00264)
custard apple (score = 0.00141)
earthstar (score = 0.00107)
Aruns-MacBook-Pro:imagenet arundas$ python classify_image.py --model_dir ~/Documents/
W tensorflow/core/framework/op_def_util.cc:332] Op BatchNormWithGlobalNormalization is deprecated. It will cease to work in GraphDef version 9. Use tf.nn.batch_normalization().
giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca (score = 0.89233)
indri, indris, Indri indri, Indri brevicaudatus (score = 0.00859)
lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens (score = 0.00264)
custard apple (score = 0.00141)
earthstar (score = 0.00107)
модель не была загружена во второй раз. Надеюсь, это поможет.
Спасибо! вторая загрузка была, вероятно, потому, что я забыл деактивировать virtualenv, в котором я находился. Но могу ли я просто сохранить два необходимых файла в/tmp/imagenet непосредственно из архива? – Woppa
спасибо @Arun, btw .. – Woppa
Да, вы можете использовать файлы, используя флаг '--model_dir'. –