1. 程式人生 > >TensorFlow Mnist資料集下載問題

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)