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

安装好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)

你可能感兴趣的:(TensorFlow mnist数据集路径 MNIST_data 数据下载问题)