在学习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)
运行效果如下