TensorFlow-显示来自MNIST数据集的图像
image
python
tensorflow
6
0

我正在尝试学习TensorFlow,并通过以下链接实现了MNIST示例: http ://openmachin.es/blog/tensorflow-mnist我希望能够实际查看训练/测试图像。所以我试图添加代码以显示第一批的第一张火车图片:

x_i = batch_xs[0]
image = tf.reshape(x_i,[28,28])

现在,因为数据为float32类型(值在[0,1]范围内),所以我尝试将其转换为uint16,然后将其编码为png以显示图像。我尝试使用tf.image.convert_image_dtype and tf.image.encode_png ,但是没有成功。你们能帮助我了解如何将原始数据转换为图像并显示图像吗?

参考资料:
Stack Overflow
收藏
评论
共 3 个回答
高赞 时间 活跃

在针对ML初学者的MNIST教程中的代码之上,您可以可视化mnist数据集中的图像:

import matplotlib.pyplot as plt
batch = mnist.train.next_batch(1)
plotData = batch[0]
plotData = plotData.reshape(28, 28)
plt.gray() # use this line if you don't want to see it in color
plt.imshow(plotData)
plt.show()

在此处输入图片说明

收藏
评论

阅读本教程后,您可以在numpy中完成所有操作,而无需TF:

import matplotlib.pyplot as plt
first_array=batch_xs[0]
#Not sure you even have to do that if you just want to visualize it
#first_array=255*first_array
#first_array=first_array.astype("uint8")
plt.imshow(first_array)
#Actually displaying the plot if you are not in interactive mode
plt.show()
#Saving plot
plt.savefig("fig.png")

您还可以使用PIL或您所使用的任何可视化工具。

收藏
评论
X = X.reshape([28, 28]);
plt.gray()
plt.imshow(X)

这可行。

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号