如何使用VGG網(wǎng)絡(luò)進行MNIST圖像分類

**1 **問題

VGG網(wǎng)絡(luò)由牛津大學(xué)的Oxford Visual Geometry Group于2015年提出。從誕生之后就收到了學(xué)界的廣泛關(guān)注。

VGG網(wǎng)絡(luò)，可以應(yīng)用在人臉識別、圖像分類等方面。VGG有兩種結(jié)構(gòu)，分別為16層和19層。具體結(jié)構(gòu)在其文獻做了詳細表述，如下圖所示。

為了學(xué)習(xí)VGG網(wǎng)絡(luò)，本組擬采用配置A在MNIST數(shù)據(jù)集上進行圖像分類實驗。

**2 **方法

首先MNIST的數(shù)據(jù)大小為28*28，需要進行resize才能作為VGG網(wǎng)絡(luò)的輸入；同時，本次實驗只需要進行10分類，因此將網(wǎng)絡(luò)本身的最后一層原做1000分類的soft-max層移除，替換為FC-10。網(wǎng)絡(luò)實現(xiàn)代碼如下：

獲取數(shù)據(jù)后進行resize操作：

實驗部分代碼如下:

實驗結(jié)果：

**3 **結(jié)語

VGG主要工作是證明了增加網(wǎng)絡(luò)的深度能夠在一定程度上影響網(wǎng)絡(luò)最終的性能，從本次實驗也可以看出，短時間少周期的訓(xùn)練并不能使得如此龐大的網(wǎng)絡(luò)擁有很好的效果，而在比較小的網(wǎng)絡(luò)如LeNet-5這樣的網(wǎng)絡(luò)上往往幾個周期就能得到較高的精度。遺憾的是因為實驗設(shè)備性能限制，網(wǎng)絡(luò)的運行速度很慢，受限于內(nèi)存大小，BatchSize的大小受限，最大只能到32，沒有充足的調(diào)整空間。