1. 程式人生 > >將mnist獲得的資料還原成圖片形式

將mnist獲得的資料還原成圖片形式

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()

效果如下: