将图像输入张量流

时间:2016-12-20 02:55:54

标签: machine-learning tensorflow mnist softmax

我是使用TensorFlow的新手。所以我在ML中为初学者尝试了MNIST教程。代码运行得很好。但是,如果我想输入我自己的图像,如果它上面有一个手写的数字,如果它预测它可能是什么数字,该怎么办?如何将自己的图像输入TensorFlow程序?

1 个答案:

答案 0 :(得分:1)

假设您正在使用this file

如果查看x,形状为[None, 784]。要提供您自己的图像,您必须将图像存储为变量(使用PIL或OpenCV或其他东西加载),展平它,将其包装在列表中,然后将其传递给{{1}中的图形看起来像这样:

feed_dict

它需要是一个28x28的图像才能使这段代码无需修改即可运行。