【发布时间】:2016-12-20 02:55:54
【问题描述】:
我是使用 TensorFlow 的新手。所以我在为初学者准备 ML 中的 MNIST 教程。代码运行得很好。但是如果我想输入一张我自己的图像,上面写着一个手写的数字,如果它预测它可能是什么数字呢?如何将自己的图像输入 TensorFlow 程序?
【问题讨论】:
标签: machine-learning tensorflow mnist softmax
我是使用 TensorFlow 的新手。所以我在为初学者准备 ML 中的 MNIST 教程。代码运行得很好。但是如果我想输入一张我自己的图像,上面写着一个手写的数字,如果它预测它可能是什么数字呢?如何将自己的图像输入 TensorFlow 程序?
【问题讨论】:
标签: machine-learning tensorflow mnist softmax
假设您使用的是this file。
如果你看x,形状是[None, 784]。要输入您自己的图像,您必须将图像存储为变量(使用 PIL 或 OpenCV 或其他方式加载它),将其展平,将其包装在列表中,并将其传递给 feed_dict 中的图形,看起来像这样:
sess.run(y_, feed_dict={x: [np.flatten(image_you_loaded_in)]})
它需要是 28x28 的图像才能使此代码无需修改即可工作。
【讨论】: