tensorflow正常加载mnist数据集

一、加载mnist数据集

    使用from tensorflow.examples.tutorials.mnist import input_data来将tensorflow自带的input_data.py导入到项目中,该文件中导入了一些模块和tensorflow自带的处理mnist数据集的方法。

    使用

import os
print(os.getcwd())

 来查看当前的工作目录,并且从这里将四个压缩文件下载下来,在当前工作目录下新建MNIST_data文件夹,将这四个压缩文件直接放入该文件夹。

 接着使用

mnist=input_data.read_data_sets('./MNIST_data/',one_hot=True)
 来将数据读出。其中tensorflow自带的方法read_data_sets()会先在指定文件夹中查看有无需要的文件,若有,则直接加载,否则会从默认的网址自动下载,但是自动下载一般都会出错,所以我们还是事先将数据集自己下载下来。

猜你喜欢

转载自blog.csdn.net/qq_26593465/article/details/81026762