上一篇筆記主要介紹了卷積神經(jīng)網(wǎng)絡(luò)相關(guān)的基礎(chǔ)知識(shí)。在本篇筆記中,將參考TensorFlow官方文檔使用mnist數(shù)據(jù)集,在TensorFlow上訓(xùn)練一個(gè)多層卷積神經(jīng)網(wǎng)絡(luò)。
下載并導(dǎo)入mnist數(shù)據(jù)集
首先,利用input_data.py來(lái)下載并導(dǎo)入mnist數(shù)據(jù)集。在這個(gè)過(guò)程中,數(shù)據(jù)集會(huì)被下載并存儲(chǔ)到名為"MNIST_data"的目錄中。
import input_data mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
其中mnist是一個(gè)輕量級(jí)的類,其中以Numpy數(shù)組的形式中存儲(chǔ)著訓(xùn)練集、驗(yàn)證集、測(cè)試集。