MNIST用matplotlib實現可視化

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np
from tensorflow.examples.tutorials.mnist import input_data

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

i = 0
# 獲取圖片
image = mnist.train.images[i]
# 將圖像數據還原成28*28的分辨率
image = image.reshape(28, 28)
# 打印對應的標籤
answer = np.argmax(mnist.train.labels[i])
plt.imshow(image,cmap='binary')
title = 'No.'+str(i-int('0'))+': '+str(answer-int('0'))
plt.title(title)
plt.show()