首页 > 代码库 > 如何显示mnist中的数据(tensroflow)
如何显示mnist中的数据(tensroflow)
在使用mnist数据集的时候，一直想看看数据中原来的图片，还有卷积层、池化层中的图片，经过不断的捣鼓，最后显示了出来。只看数据集中的图片用如下代码就好了：
1 import tensorflow.examples.tutorials.mnist.input_data as input_data 2 import numpy as np 3 import matplotlib.pyplot as plt 4 import pylab 5 6 mnist = input_data.read_data_sets("MNIST_data/", one_hot=True) 7 8 batch_xs, batch_ys = mnist.train.next_batch(100) 9 for one_pic_vic in batch_xs: 10 one_pic_arr = np.reshape(one_pic_vic,(28,28)) 11 plt.imshow(one_pic_arr) 12 pylab.show()
batch_xs的Size是(100,784)，其中100是由batch大小决定，mnist中的每张图片本来的大小是28x28的，然后数据集中存成了1x784，所以batch_xs对应100张图片。上面的代码通过reshape把图片又转成了28x28，然后就可以通过plt.imshow()显示出来:
如果要看卷积神经网络中的卷积层、池化层，也可以用类似的方法，不过要先使用sess.run()方法来提取出来卷积层、池化层，因为图像比较多，所以就用plt.imsave()来保存到文件中。
如何显示mnist中的数据(tensroflow)
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。