Tensorflow学习--识别mnist数据集
在学习Tensorflow时,很多例子都是采用的mnist数据集,首先是下载mnist数据集,下载地址如下:
http://www.tensorfly.cn/tfdoc/tutorials/mnist_download.html
将链接中四个压缩文件都下载下来,
如下图所示
然后四个压缩文件都放入一个新建的MNIST_data文件夹,不用解压直接放入文件夹就好了。
这里的数据集文件夹和写代码的文件是同一路径,方便读取,如果不是同一路径,在读取数据集的时候记得写全路径。如:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/",one_hot=True)
print("输入数据:",mnist.train.images)
print("输入数据打印shape:",mnist.train.images.shape)
import pylab#图形打印类
im = mnist.train.images[1]
im = im.reshape(-1,28)
pylab.imshow(im)
pylab.show()
print("输入数据打印shape",mnist.test.images.shape)
print("输入数据打印shape",mnist.validation.images.shape)
运行效果如下