基于卷積神經(jīng)網(wǎng)絡的圖像識別教學實驗

祝世平， 周富強， 魏新國， 溫思寒

(北京航空航天大學 儀器科學與光電工程學院， 北京 100191)

基于卷積神經(jīng)網(wǎng)絡的圖像識別教學實驗

祝世平， 周富強， 魏新國， 溫思寒

(北京航空航天大學 儀器科學與光電工程學院， 北京 100191)

圖像識別是“圖像處理”教學中的重要內容。本文在Linux環(huán)境下使用iTorch notebook可視化界面利用卷積神經(jīng)網(wǎng)絡實現(xiàn)mnist手寫數(shù)字體的準確識別，并詳細介紹卷積神經(jīng)網(wǎng)絡的原理，給出直觀的實驗結果。教學實踐表明，通過具有應用性和趣味性的實驗可以提高學生的積極性，加深對課程理論的認知，培養(yǎng)其分析問題和解決問題的能力。

圖像識別；卷積神經(jīng)網(wǎng)絡；iTorch

0 引言

圖像識別是“數(shù)字圖像處理”課程的重要內容，隨著圖像處理技術的迅速發(fā)展，圖像識別技術的應用領域越來越廣泛[1]。其中，好的識別技術是關鍵所在，怎么樣提高識別率和識別的速度意義重大，直接關系到圖像識別的實用性和安全性。目前，圖像識別方法大多采用人工方式提取特征，不僅費時費力，而且提取困難；而深度學習是一種非監(jiān)督學習，學習過程中可以不知道樣本的標簽值，整個過程無需人工參與也能提取到好的特征。作為實現(xiàn)深度學習的一項重要技術，卷積神經(jīng)網(wǎng)絡(CNN)成功訓練了首個深層神經(jīng)網(wǎng)絡結構，已在圖像識別、語音識別等領域取得卓越成績[2]。另外，在Google公司的alphGo運用深度學習的方法戰(zhàn)勝圍棋高手李世石后，人們對深度學習的好奇掀起了高潮，故為了更好地滿足學生的求知欲，激發(fā)學生的積極性，本課程設計了基于卷積神經(jīng)網(wǎng)絡的圖像識別實驗，并在iTorch notebook可視化界面下訓練MNIST手寫數(shù)據(jù)集，測試其識別準確度。通過直觀的實驗結果，使學生在實驗中達到以下鍛煉：

(1)熟悉Linux開發(fā)環(huán)境，并掌握對Torch，iTorch的安裝及使用；

(2)了解深度學習，并理解卷積神經(jīng)網(wǎng)絡的概念，掌握其實現(xiàn)流程；

(3)更加深刻的理解圖像識別的原理，并對卷積神經(jīng)網(wǎng)絡技術在圖像識別上的應用有所掌握。

另外，實驗提供的可視化平臺，使學生可以更直觀地對程序進行調試及運行，提高了其動手實踐和分析與解決問題的能力。

1 實驗環(huán)境配置

本實驗使用的 iTorch notebook 是一個通過Web界面實現(xiàn)交互的shell，其中iTorch 是 Torch 的 iPython 內核。

首先，我們需要一臺配置好Python和Torch7的Linux系統(tǒng)，再針對iTorch notebook配置iTorch，iTorch是Torch里面的一個包，它能很輕松地顯示圖片、視頻和音頻等等，但是需要和ipython搭配使用，故需要安裝iPython。

打開終端，輸入：

wget http://ipython.scipy.org/dist/ipython-0.8.4.tar.gz

下載完畢后，輸入解壓縮命令：tar zxvf ipython-0.8.4.tar.gz

切換到解壓縮的目錄：cd ipython-0.8.4

輸入命令安裝：python setup.py install

但此時還是無法使用，需要我們再安裝ipython-notebook，安裝命令如下：sudo apt-get install ipython-notebook

然后，在終端輸入：itorch notebook，在瀏覽器中便會彈出iTorch notebook的Web界面，如圖1所示。

圖1 iTorch notebook的Web界面

點擊右上角的new notebook建立一個新的筆記本，界面如圖2所示。

圖2 建立新筆記本

并在In[]:命令行輸入命令測試，便輸出如下圖3的界面。

圖3 命令測試結果

由圖3可看出，該界面能夠直觀顯示出我們的輸入，并能分層次輸入，便于學生對程序更好的調試及理解。

另外，由于本次實驗需要對圖像進行顯示，故需要安裝image庫，在運行itorch notebook 前在終端鍵入：luarocks install image進行安裝。

最后，本文是針對MNIST數(shù)據(jù)庫中的數(shù)字進行訓練及識別，故需要對其進行安裝，在運行itorch notebook前在終端鍵入：luarocks install mnist完成安裝。

MNIST是一個手寫數(shù)字數(shù)據(jù)庫，它有60000個訓練樣本集和10000個測試樣本集。它是NIST數(shù)據(jù)庫的一個子集。數(shù)據(jù)庫存儲并不是標準的圖像格式，這些圖像數(shù)據(jù)都保存在二進制文件中，所有的圖片都經(jīng)過了尺寸標準化和中心化，圖片的大小固定為28×28。

2 實驗設計

為了加強學生對卷積神經(jīng)網(wǎng)絡在圖像識別的理解和應用，提高學生運用卷積神經(jīng)網(wǎng)絡等深度學習方法來解決實際生活中圖像識別的能力，本實驗要求學生自行搭建程序運行所需環(huán)境，熟悉對Linux系統(tǒng)的使用，并給出已經(jīng)訓練好的訓練集，讓學生自己對測試集中的數(shù)據(jù)進行測試，給出其準確度，并隨機顯示圖片驗證其正確性。

同時，為了讓學生更快速的進入實驗狀態(tài)，發(fā)揮自己的動手實踐能力，我們給出卷積神經(jīng)網(wǎng)絡的結構，并給出該識別過程中最主要的模型建立流程。

2.1 卷積神經(jīng)網(wǎng)絡的網(wǎng)絡結構

卷積神經(jīng)網(wǎng)絡CNN(Convolutional Neural Networks)是一種為了處理二維輸入數(shù)據(jù)而特殊設計的多層人工神經(jīng)網(wǎng)絡，網(wǎng)絡中的每層都由多個二維平面組成，而每個平面由多個獨立的神經(jīng)元組成[3-5]。

如圖4示，這是一個簡單的卷積神經(jīng)網(wǎng)絡 CNN。

圖4 簡單的CNN

輸入圖像通過和三個可訓練的濾波器和可加偏置進行卷積，卷積后在C1層產(chǎn)生三個特征映射圖。然后特征映射圖中每組的四個像素再進行求最大值，加權值，加偏置，通過一個ReLu函數(shù)得到三個S2層的特征映射圖。這些映射圖再經(jīng)過濾波得到C3層。這個層級結構再和S2一樣產(chǎn)生S4。最終，這些像素值被光柵化，并連接成一個向量輸入到傳統(tǒng)的神經(jīng)網(wǎng)絡，得到輸出。

一般地，C層為特征提取層，每個神經(jīng)元的輸入與前一層的局部感受野相連，并提取該局部的特征，一旦該局部特征被提取后，它與其他特征間的位置關系也隨之確定下來；S層是特征映射層，網(wǎng)絡的每個計算層由多個特征映射組成，每個特征映射為一個平面，平面上所有神經(jīng)元的權值相等。特征映射結構采用影響函數(shù)核小的sigmoid函數(shù)作為卷積網(wǎng)絡的激活函數(shù)，使得特征映射具有位移不變性。

此外，由于一個映射面上的神經(jīng)元共享權值，因而減少了網(wǎng)絡自由參數(shù)的個數(shù)，降低了網(wǎng)絡參數(shù)選擇的復雜度。卷積神經(jīng)網(wǎng)絡中的每一個特征提取層(C-層)都緊跟著一個用來求局部平均與二次提取的計算層(S-層)，這種特有的兩次特征提取結構使網(wǎng)絡在識別時對輸入樣本有較高的畸變容忍能力。

2.2 模型建立流程

基于深度學習的圖像識別模型流程主要是：把圖像輸入到神經(jīng)網(wǎng)絡中，利用深度學習的前向傳播和反向傳播誤差等算法來最小化損失函數(shù)，更新權值后，得到一個較優(yōu)的識別模型，然后利用此模型對新的圖像來進行識別。

在該基于卷積神經(jīng)網(wǎng)絡的圖像識別實驗中，最核心的部分就是此模型的建立。

首先，我們需要首先建立一個容器用來存放各種模塊：

model = nn.Sequential()

然后，放入一個reshape模塊。因為mnist庫的原始圖片是儲存為1列728個像素的。我們需要把它們變成1通道28×28的一個方形圖片。

model:add(nn.Reshape(1, 28, 28))

接下來要把圖片的每個像素除以256再乘以3.2，也就是把像素的取值歸一化到0至3.2之間。

model:add(nn.MulConstant(1/256.0×3.2))

然后是第一個卷積層，它的參數(shù)按順序分別代表：輸入圖像是1通道，卷積核數(shù)量20，卷積核大小5×5，卷積步長1×1，圖像留邊0×0。

model:add(nn.SpatialConvolutionMM(1, 20, 5, 5, 1, 1, 0, 0))

一個池化層，它的參數(shù)按順序分別代表：池化大小2×2，步長2×2，圖像留邊0×0。

model:add(nn.SpatialMaxPooling(2, 2 , 2, 2, 0, 0))

再接一個卷積層和一個池化層，由于上一個卷積層的核的數(shù)量是20，所以這時輸入圖像的通道個數(shù)為20。

model:add(nn.SpatialConvolutionMM(20, 50, 5, 5, 1, 1, 0, 0))

model:add(nn.SpatialMaxPooling(2, 2 , 2, 2, 0, 0))

在接入全連接層之前，我們需要把數(shù)據(jù)重新排成一列，所以有需要一個reshape模塊。

model:add(nn.Reshape(4×4×50))

這個參數(shù)為什么是4×4×50，即800呢？原因是：我們的輸入是1通道28×28的圖像，經(jīng)過第一個卷積層之后變成了20通道24×24的圖像。又經(jīng)過池化層，圖像尺寸縮小一半，變?yōu)?0通道12×12。通過第二個卷積層，變?yōu)?0通道8×8的圖像，又經(jīng)過池化層縮小一半，變?yōu)?0通道4×4的圖像。所以這其中的像素一共有4×4×50=800個。

接下來是第一個全連接層。輸入為4×4×50=800，輸出為500。

model:add(nn.Linear(4×4×50, 500))

兩個全連接層之間有一個ReLU激活層。

model:add(nn.ReLU())

然后是第二個全連接層，輸入是500，輸出是10，也就代表了10個數(shù)字的輸出結果，哪個節(jié)點的響應高，結果就定為對應的數(shù)字。

model:add(nn.Linear(500, 10))

最后是一個LogSoftMax層，用來把上一層的響應歸一化到0至1之間。

model:add(nn.LogSoftMax())

至此，模型的建立就完成了。我們還需要一個判定標準。由于我們這一次是要解決分類問題，一般使用nn.ClassNLLCriterion這種類型的標準(Negative Log Likelihood)。

criterion = nn.ClassNLLCriterion()

到這里，網(wǎng)絡模型的部分就都已經(jīng)完成了，其具體流程圖如下圖5。

圖5 網(wǎng)絡模型流程圖

3 實驗結果

本次實驗最終確定神經(jīng)網(wǎng)絡的迭代次數(shù)(iteration)為30，其正確率如圖6所示，可達到98%以上，相比較其他算法，錯誤率大大減少，且由觀察可知，只需迭代30次，準確率便可穩(wěn)定，縮短了訓練時間，提高了運行速度。

圖6 6次迭代圖像識別正確率

針對訓練好的模型，我們從測試集中隨機選取圖片進行驗證，其結果示于圖7和8，無論什么形式的手寫體,該算法都能夠對其正確的識別。

圖7圖像識別算例1測試結果 圖8 圖像識別算例2測試結果

4 結語

本文詳細討論了基于卷積神經(jīng)網(wǎng)絡的圖像識別實驗的原理和流程，并以iTorch noteboook的可視化界面直觀的顯示出其準確性。而且，通過讓學生自行搭建程序運行所需環(huán)境，自行編程對測試集中的數(shù)據(jù)進行測試，并隨機顯示圖片驗證其正確性，大大提高了學生的編程能力，加深了學生對深度學習的了解，讓學生更加全面的理解卷積神經(jīng)網(wǎng)絡在圖像識別中的應用。

[1] 鄧柳, 汪子杰. 基于深度卷積神經(jīng)網(wǎng)絡的車牌識別研究[J]. 成都:計算機應用研究, 2016, 33(3): 930-932.

[2] 徐彩云. 圖像識別技術研究綜述[J]. 菏澤:電腦知識與技術, 2013, 9(10): 2446-2447.

[3] 盧宏濤, 張秦川. 深度卷積神經(jīng)網(wǎng)絡在計算機視覺中的應用研究綜述[J]. 南京:數(shù)據(jù)采集與處理, 2016, 31(1): 1-17.

[4] 張文達, 許悅雷, 倪嘉成, 馬時平, 史鶴歡. 基于多尺度分塊卷積神經(jīng)網(wǎng)絡的圖像目標識別算法[J]. 成都:計算機應用, 2016, 36(4): 1033-1038.

[5] 范哲意, 曾亞軍, 蔣姣, 翁澍沁, 劉志文. 視頻人臉識別實驗平臺的設計與實現(xiàn)[J]. 北京:實驗技術與管理, 2016, 33(3): 159-161, 165.

Image Recognition Teaching Experiment Based on Convolutional Neural Network

ZHU Shi-ping, ZHOU Fu-qiang, WEI Xin-guo, WEN Si-han

(SchoolofInstrumentationScienceandOptoelectronicsEngineering,BeihangUniversity,Beijing100191,China)

Image recognition is the important content of Image Processing course. In this paper, under the environment of Linux, we realize the accurate handwritten digit recognition from the data of mnist by using a visual interface called itorch notebook, introduce the principle of the convolutional neural network in detail, and give an intuitive experimental results. The teaching experience shows that such kind of experiment which embodies interestingness and practical application has the advantage of increasing the students′ interest in the course, deepen the cognition of curriculum theory and cultivate their ability to analyze and solve problems.

image recognition; convolutional neural network; iTorch

2016-07-05;

2016-11- 11

北京航空航天大學儀器科學與光電工程學院教改項目

祝世平(1970-)，男，副教授，主要從事圖像處理教學和視頻壓縮、計算機視覺、機器視覺在精密測量中的應用等方面的研究，E-mail：spzhu@163.com

G426

A

1008-0686(2017)04-0124-04