TensorFlow Mnist数据集下载问题

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

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

一、实际运行时根本无法通过网络下载,原因是需要科学上网。解决方案就是手工下载数据,然后直接导入使用。

因此我们可以去:http://yann.lecun.com/exdb/mnist/ 下载所需的4个文件(可以参考TensorFlow中文社区数据下载这一节内容),然后放到当前目录MNIST_data下,即可正常使用。另外,MNIST_data是本地地址,可以自定义路径。

二、测试是否可以正常使用

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

猜你喜欢

转载自blog.csdn.net/qq_34333481/article/details/83927849