python - 如何在 TensorFlow 中使用模型 Inception.tgz？

Question

我刚刚从tensorflow.org下载了inception.tgz文件，地址为http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz 。但是，我不知道应该在哪里提取它。

此外，当我使用 models/image/imagenet/classify_image.py 脚本获取模型时，重新启动后模型未保存，因此我必须通过相同的脚本再次下载它。我有时需要在没有连接到互联网的情况下使用它，因此每次需要分类时都下载模型对我来说并不理想。我怎样才能一劳永逸地坚持这个模型？

另外，我如何使用 .tgz 初始模型？

Answer 1

我无法对您的问题发表评论，因为我还没有足够的学分。那么让我给你一个通用的答案。

1. 您提到的 inception-2015-12-05.tgz 文件包含您需要的两个文件:

   a) imagenet_comp_graph_label_strings.txt

   b)tensorflow_inception_graph.pb

还有一个您也不需要的许可证文件。这两个文件可让您对图像进行预测。

您提到的部分重启后未保存模型，因此我必须通过相同的脚本再次下载它引起了我的兴趣。我从来没有遇到过这样的问题。现在就试试这个:

• 在您选择的位置创建一个文件夹。说~/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)

第二次未下载模型。希望这会有所帮助。