pip install scipy==1.1.0
代码
import tensorflow as tf
import scipy.misc
import os
from PIL import Image
from tensorflow.examples.tutorials.mnist import input_data
mnist=input_data.read_data_sets("MNIST_data",one_hot=True)
print(mnist.train.labels[0,:])
print(mnist.train.labels[1,:])
save_dir='MNIST_data/raw/'
if os.path.exists(save_dir) is False:
os.mkdir(save_dir)
import numpy as np
for i in range(2):
image_array=mnist.train.images[i,:]
image_array=image_array.reshape(28,28)
filename=save_dir+'%d.jpg' %i
img = scipy.misc.toimage( image_array)#.convert('RGB')
img.save(filename)
one_hot_label=mnist.train.labels[i,:]
label = np.argmax(one_hot_label)
print("%d.jpg,%d"%(i,label))