MNIST用matplotlib实现可视化

最新推荐文章于 2024-04-11 08:57:30 发布

Neo__Z

最新推荐文章于 2024-04-11 08:57:30 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： --------Tensorflow 学习--------

本文链接：https://blog.csdn.net/cdsszjj/article/details/102810421

--------Tensorflow 学习-------- 专栏收录该内容

5 篇文章

订阅专栏

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()