from tensorflow.examples.tutorials.mnist import input_data
import scipy.misc
import os
mnist = input_data.read_data_sets('MNIST_data/',one_hot=True)
save_dir = 'MNIST_data/raw/'
if os.path.exists(save_dir) is False:
os.makedirs(save_dir)
for i in range(20):
image_array = mnist.train.images[i,:]
image_array = image_array.reshape(28, 28)
filename = save_dir + 'mnist_data_%d.jpg' % i
scipy.misc.toimage(image_array,cmin=0.0,cmax=1.0).save(filename)
相关文章
- 11-07保存mnist前20张图片