保存mnist前20张图片

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)
上一篇:tf.reset_default_graph


下一篇:pandas.read_csv与data.to_csv中文件路径