2016-09-24 2 views
1

Я только что загрузил файл inception.tgz с сайта tensorflow.org по адресу http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz. Но я не знаю, где я должен это извлечь.Как использовать модель Inception.tgz в TensorFlow?

Кроме того, когда я использовал сценарий models/image/imagenet/classify_image.py, чтобы получить модель, модель не была сохранена после перезагрузки, поэтому мне пришлось загрузить ее снова через тот же скрипт. Мне нужно использовать его порой, я не подключен к Интернету, поэтому загрузка модели каждый раз, когда мне нужно классифицировать, не идеальна для меня. Как я могу сохранить модель раз и навсегда?

Также, как я могу использовать начальную модель .tgz?

ответ

2

Я не могу сделать комментарий к вашему Вопрос, так как у меня пока нет кредитов. Поэтому позвольте мне дать вам общий ответ.

  1. Файл inception-2015-12-05.tgz вы упомянули содержит два файла, которые вам требуется:

    а) imagenet_comp_graph_label_strings.txt

    б) tensorflow_inception_graph.pb

Существует файл лицензии, который вам также не потребуется. Эти два файла позволят вам делать прогнозы на изображения.

  1. Часть, в которой вы упоминали 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) 

модель не была загружена во второй раз. Надеюсь, это поможет.

+0

Спасибо! вторая загрузка была, вероятно, потому, что я забыл деактивировать virtualenv, в котором я находился. Но могу ли я просто сохранить два необходимых файла в/tmp/imagenet непосредственно из архива? – Woppa

+0

спасибо @Arun, btw .. – Woppa

+0

Да, вы можете использовать файлы, используя флаг '--model_dir'. –

1

По умолчанию изображения модель получает загружена в/TMP/imagenet, но вы можете установить собственную папку, передав в параметре --model_dir командной строки для classify_image.py: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/models/image/imagenet/classify_image.py#L56

+0

спасибо @PeteWarden. Я работаю над этим. Я только начинаю ML и мало знаю, но очень возбужден. Еще раз спасибо – Woppa

0

потому что он сохраняется в каталоге temp, каждый раз, когда вы закрываете машину, она удаляется. попробуйте сохранить файл в каталоге с помощью параметра --model_dir.

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