MNIST是一个手写数字数据集，里面所包含的数据元素是类似于一个1×784矩阵和1×10矩阵的结构，那么，如何将这些数据元素转化为更加直观的图像呢？通过以下python代码，可以实现。
1 from PIL import Image 2 from tensorflow.examples.tutorials.mnist import input_data 3 import numpy as np 4 import matplotlib.pyplot as plt 5 6 #获取数据集 7 mnist = input_data.read_data_sets("MNIST_data/",one_hot=True) 8 #fdata为图像数据，label为该图像的标签 9 fdata,label = mnist.train.next_batch(1) 10 #将1*784矩阵变为28*28的矩阵 11 data = np.matrix(fdata).reshape(28,28) 12 data = data*255 13 #矩阵转图像 14 new_im = Image.fromarray(data.astype(np.uint8)) 15 new_im.show()
效果如下：
