TENSorflow中已保存模型的推理文档

时间:2017-01-31 09:18:30

标签: tensorflow

Multilayer CNN for MNIST Digit Recognition所述,我为MNIST图像识别数据集创建了一个CNN。我不想在同一个脚本中测试它,所以我训练了模型,保存了检查点文件和图形结构,然后使用freeze_graph.py脚本。我把它冻结成一个.pb文件。现在使用这个.pb文件,我想推断出MNIST测试数据集,并检查模型的准确性,但没有明确的文档说明如何做到这一点。有一个例子写在examples / label_images目录中,但也是在C ++和初始网络中。我已经解析了.pb文件并创建了一个tf.Graph对象。还有为什么不能提供单个推理脚本,因为可以从保存在.pb文件中的图形中读取整个图形结构。

0 个答案:

没有答案