TensorFlowのtutorialにあるmnistデータをImageで表示する

TensorFlowTutorialMinistDataViewer.py

%matplotlib inline
import tensorflow.examples.tutorials.mnist.input_data as input_data
import numpy
import matplotlib.pyplot as pyplot
import matplotlib.cm as cm
from PIL import Image


mnist = input_data.read_data_sets('./MNIST_data', one_hot=True)

data = mnist.train
pyplot.imshow(Image.fromarray(numpy
                              .uint8(data.images[0] * 255)
                              .reshape(28, 28))
              , cmap = cm.Greys_r)
print('Number:{0}'.format(data.labels[0].argmax()))