Tensorflow - 按照教程中的描述构建CNN模型

时间:2017-05-13 15:56:54

标签: python-3.x tensorflow mnist

我刚刚完成了“TF图层指南:为MNIST数据集构建卷积神经网络”的实现。训练模型成功运行,准确率为97.3%。

但是,本教程未提及如何使用这种新训练的模型来提供自己的图像并查看预测。有谁知道如何使用训练模型的输出来进行预测?我在tmp / mnist_convnet_model $文件夹中看到,有一些输出文件,如.pbtxt,元文件和索引文件。但我无法找到使用它们对我自己的图像进行预测的说明。

1 个答案:

答案 0 :(得分:0)

y_pred = tf.nn.softmax(your_final_layer)
y_pred_cls = tf.argmax(y_pred, dimension=1)

和预测

feed_dict = {x: [your_image]}
classification = tf.run(y_pred_cls, feed_dict)
print classification

这几乎适用于您创建的任何模型