如何评估TensorFlow模型给出的.pb文件?

时间:2016-07-21 11:47:20

标签: tensorflow

我使用以下命令下载了Tensorflow的Inception_v3模型:

curl http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz -o /tmp/inceptionv3.tgz
tar xzf /tmp/inceptionv3.tgz -C /tmp/

现在我有一个classify_image_graph_def.pb文件,我相信它是模型。

我的问题是,如何针对ImageNet 2012数据评估此模型?是否有任何脚本,或者我是否必须编写一些python代码?

1 个答案:

答案 0 :(得分:0)

TensorFlow GitHub存储库中的classify_image.py脚本使用pre-trained Inception model对任意JPEG图像执行分类。您可以调整此脚本以根据ImageNet 2012数据对其进行评估。