TensorFlow mnist数据集路径 MNIST_data 数据下载问题

左心房为你撑大大i 提交于 2019-12-05 07:54:28

安装好TensorFlow后,按教程输入如下命令时,会出现不能下载数据的问题。

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot = True)

问题一般是连接超时或者下载失败,这是因为下载地址默认为:https://storage.googleapis.com/cvdf-datasets/mnist/,国内不能直接访问,因此我们可以去:http://yann.lecun.com/exdb/mnist/ 下载所需的4个文件(可以参考TensorFlow中文社区数据下载这一节内容),然后放到当前目录MNIST_data下,即可正常使用。另外,MNIST_data是本地地址,可以自定义路径。

执行下面测试语句测试是否正常

print(mnist.train.images.shape, mnist.train.labels.shape)

结果:

(55000, 784) (55000, 10)

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!