在针对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)

这可行。