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.youkuaiyun.com/cdsszjj/article/details/102810421

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

5 篇文章

订阅专栏

本文通过使用TensorFlow读取MNIST数据集，并利用Matplotlib进行图像展示，实现了对MNIST手写数字数据集的可视化。文章展示了如何从数据集中获取一张图片，将其还原为28x28像素的分辨率，并打印出对应的标签。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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