TensorFlow官方文檔MNIST 數據下載

MNIST 數據下載

源碼: tensorflow/g3doc/tutorials/mnist/

本教程的目標是展示如何下載用於手寫數字分類問題所要用到的(經典)MNIST數據集。

教程 文件

本教程需要使用以下文件:

文件目的
input_data.py下載用於訓練和測試的MNIST數據集的源碼

準備數據

MNIST是在機器學習領域中的一個經典問題。該問題解決的是把28x28像素的灰度手寫數字圖片識別爲相應的數字,其中數字的範圍從0到9.

艾伯特(http://www.aibbt.com/)國內第一家人工智能門戶

TensorFlow 官方文檔中文版

更多詳情, 請參考 Yann LeCun's MNIST page 或 Chris Olah's visualizations of MNIST.

下載

Yann LeCun's MNIST page 也提供了訓練集與測試集數據的下載。

文件內容
train-images-idx3-ubyte.gz訓練集圖片 - 55000 張 訓練圖片, 5000 張 驗證圖片
train-labels-idx1-ubyte.gz訓練集圖片對應的數字標籤
t10k-images-idx3-ubyte.gz測試集圖片 - 10000 張 圖片
t10k-labels-idx1-ubyte.gz測試集圖片對應的數字標籤

在 input_data.py 文件中, maybe_download() 函數可以確保這些訓練數據下載到本地文件夾中。

文件夾的名字在 fully_connected_feed.py 文件的頂部由一個標記變量指定,你可以根據自己的需要進行修改。

解壓 與 重構

這些文件本身並沒有使用標準的圖片格式儲存,並且需要使用input_data.py文件中extract_images() 和extract_labels()函數來手動解壓(頁面中有相關說明)。

圖片數據將被解壓成2維的tensor:[image index, pixel index] 其中每一項表示某一圖片中特定像素的強度值, 範圍從 [0, 255] 到 [-0.5, 0.5]。 "image index"代表數據集中圖片的編號, 從0到數據集的上限值。"pixel index"代表該圖片中像素點得個數, 從0到圖片的像素上限值。

train-*開頭的文件中包括60000個樣本,其中分割出55000個樣本作爲訓練集,其餘的5000個樣本作爲驗證集。因爲所有數據集中28x28像素的灰度圖片的尺寸爲784,所以訓練集輸出的tensor格式爲[55000, 784]

數字標籤數據被解壓稱1維的tensor: [image index],它定義了每個樣本數值的類別分類。對於訓練集的標籤來說,這個數據規模就是:[55000]

數據集 對象

底層的源碼將會執行下載、解壓、重構圖片和標籤數據來組成以下的數據集對象:

數據集目的
data_sets.train55000 組 圖片和標籤, 用於訓練。
data_sets.validation5000 組 圖片和標籤, 用於迭代驗證訓練的準確性。
data_sets.test10000 組 圖片和標籤, 用於最終測試訓練的準確性。

執行read_data_sets()函數將會返回一個DataSet實例,其中包含了以上三個數據集。函數DataSet.next_batch()是用於獲取以batch_size爲大小的一個元組,其中包含了一組圖片和標籤,該元組會被用於當前的TensorFlow運算會話中。

images_feed, labels_feed = data_set.next_batch(FLAGS.batch_size)

 http://www.aibbt.com/a/16370.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章