基于深度學(xué)習(xí)的車標識別算法的研究與實現(xiàn)

陳小娥，楊薇薇

(福建船政交通職業(yè)學(xué)院信息工程系，福州 350007)

基于深度學(xué)習(xí)的車標識別算法的研究與實現(xiàn)

陳小娥，楊薇薇

(福建船政交通職業(yè)學(xué)院信息工程系，福州 350007)

依據(jù)深度學(xué)習(xí)算法可以自主進行特征學(xué)習(xí)和識別的特點，提出一種基于Keras深度學(xué)習(xí)框架的車標識別算法。將該方法將ResNet50和Inception V3兩模型相結(jié)合構(gòu)建訓(xùn)練網(wǎng)絡(luò)，以此建立分類器對車標圖像進行識別。實驗結(jié)果表明，該方法與傳統(tǒng)的卷積神經(jīng)網(wǎng)絡(luò)分類進行比對，提高了車標識別的準確率。

深度學(xué)習(xí)；Keras框架；車標識別；特征提取

0 引言

隨著我國社會經(jīng)濟的飛速發(fā)展，人們的物質(zhì)生活水平越來越高，道路上機動車的數(shù)量也越來越多。汽車使人們的生活越來越便捷，但與此同時，隨著汽車數(shù)量的增多，也產(chǎn)生了一些不利影響，如道路擁堵、交通事故頻發(fā)等，給智能交通管理系統(tǒng)帶來了巨大的挑戰(zhàn)。

車輛識別是智能交通系統(tǒng)中的重要組成部分，目前車輛識別中的車牌識別技術(shù)已相對成熟，然而近幾年套牌、車牌磨損以及車牌遮擋等現(xiàn)象的出現(xiàn)，需要我們進一步提高車輛識別技術(shù)。于是，車標作為車輛的一個重要特征，可以輔助人們進行車輛識別。因為車標是車輛所有特征中最難以更改的重要標志之一。將車標作為研究對象，對車標進行識別，然后再與車牌識別相互結(jié)合，可進一步提高車輛識別的準確性。車標識別技術(shù)為智能交通系統(tǒng)的發(fā)展和完善提供了一種新的理念和方向，成為近幾年車輛識別中的研究熱點，將對車輛識別技術(shù)研究具有重大的推動作用。

關(guān)于車標識別，國內(nèi)外已有一些學(xué)者進行了相關(guān)研究。傳統(tǒng)的方法主要以分析特征提取方法為主，然后采用一些分類工具進行分類識別，如文獻[1]提出了一種基于SIFT描述子的車標識別算法，SIFT特征對旋轉(zhuǎn)、尺度和一定視角與光照變化等圖像變化都具有不變性，可以準確地表示車標圖像，但計算量比較大。文獻[2-3]通過計算圖像的方向梯度直方圖作為特征值，然后利用SVM作為分類器來實現(xiàn)車標識別；文獻[4]通過Modest AdaBoost算法檢測車標的位置，接著利用徑向Tchebichef矩進行車標識別。以上這些算法的車標特征均是通過人工提取獲得，特征選取得好壞直接決定了識別準確率的高低，有些圖像的特征不好表達，操作起來也相對復(fù)雜。

近年來，深度結(jié)構(gòu)學(xué)習(xí)(又稱為深度學(xué)習(xí)或多層學(xué)習(xí))作為機器學(xué)習(xí)研究的一個新的領(lǐng)域，深受人們的關(guān)注，深度學(xué)習(xí)是一種通過多層表示來對數(shù)據(jù)之間的復(fù)雜關(guān)系進行建模的算法，該算法可以有效利用復(fù)雜的非線性函數(shù)和非線性的復(fù)合函數(shù)來學(xué)習(xí)分布和分層的特征表示。文獻[5]采用了深度學(xué)習(xí)的方法，通過卷積神經(jīng)網(wǎng)絡(luò)自主選取車標特征進行學(xué)習(xí)和識別。

本文提出一種基于Keras深度學(xué)習(xí)框架的車標識別算法。該方法通過ResNet50和Inception V3兩模型導(dǎo)出特征向量，并將它們合并成一條特征向量進行訓(xùn)練建模，獲得分類器，然后通過實驗，將該算法與傳統(tǒng)卷積神經(jīng)網(wǎng)絡(luò)分類器進行對比，分析它們識別的錯誤率情況。

1 深度學(xué)習(xí)

2006年Hinton[6-7]在研討會上的兩篇論文中介紹了一種高效的無監(jiān)督學(xué)習(xí)算法，提出了“多隱層的人工神經(jīng)網(wǎng)絡(luò)具有優(yōu)異的特征學(xué)習(xí)能力”這一觀點。深度學(xué)習(xí)技術(shù)分為深度判別式模型(如深度神經(jīng)網(wǎng)絡(luò)DNN、遞歸神經(jīng)網(wǎng)絡(luò)RNN、卷積神經(jīng)網(wǎng)絡(luò)CNN等)和生成式/無監(jiān)督模型(如受限玻爾茲曼機RBM、深度置信網(wǎng)絡(luò)DBN、深度玻爾茲曼機DBM、正則化的自編碼器等)。CNN可以以圖像作為輸入，在計算機視覺或者圖像識別任務(wù)中非常有效。

1.1 傳統(tǒng)卷積神經(jīng)網(wǎng)絡(luò)

深度學(xué)習(xí)在目標識別中的最初應(yīng)用由Y.LeCun等[8]人提出，圖1展示了文獻[8]中所描述的CNN的基本結(jié)構(gòu)。CNN的深度結(jié)構(gòu)中，每個模塊都是由卷積層(Convolution)和池化層(Max Pooling)組成。這些模塊通常是逐個疊加而成，卷積層共享權(quán)值，池化層對卷積層的輸出進行降采樣，從而減少下面一層的數(shù)據(jù)率。由此可以把圖像作為直接的輸入，省去了煩瑣的特征提取過程。

圖1 初始卷積神經(jīng)網(wǎng)絡(luò)基本結(jié)構(gòu)

1.2 ResNet50模型

文獻[9]在現(xiàn)有訓(xùn)練深度網(wǎng)絡(luò)的基礎(chǔ)上，提出了一種減輕網(wǎng)絡(luò)訓(xùn)練負擔(dān)的殘差學(xué)習(xí)框架，由此解決精準度下降問題。深度殘差網(wǎng)絡(luò)(ResNet)的出現(xiàn)為訓(xùn)練更深的網(wǎng)絡(luò)提供了方法，其主要特色是跨層連接。讓這些層不是通過每一個堆疊層直接適合一個所需的底層映射，而是適合殘差映射。擬合殘差由圖2的跨層連接所示，網(wǎng)絡(luò)試圖將x映射為F(x)+x，那么網(wǎng)絡(luò)的映射F(x)自然就趨向于F(x)=0。

1.3 Inception V3模型

Christian等[10]人發(fā)布了一個新版本的GoogleNet(Inception V3)模塊和相應(yīng)的架構(gòu)，并且更好地解釋了原來的GoogleNet架構(gòu)。GoogleNet的原始思想包括通過構(gòu)建平衡深度和寬度的網(wǎng)絡(luò)，最大化網(wǎng)絡(luò)的信息流，在進入池化層之前增加特征映射；當網(wǎng)絡(luò)層數(shù)深度增加時，特征的數(shù)量或?qū)拥膶挾纫蚕鄬?yīng)地增加；在每一層通過增加寬度以增加下一層之前的特征的組合；大卷積分解為小卷積，如圖3所示用兩層3×3卷積核代替原本5×5卷積核的Inception模塊示意圖[10]。

圖2 殘差學(xué)習(xí)：一個構(gòu)建模塊

圖3 Inception模塊示意圖(用2個3×3卷積核代替1個5×5)

2 車標識別

2.1 系統(tǒng)的構(gòu)建

車標識別系統(tǒng)主要分兩部分構(gòu)成：一部分是訓(xùn)練模塊，一部分是識別模塊，如圖4所示。訓(xùn)練模塊包括訓(xùn)練圖片的輸入、深度網(wǎng)絡(luò)的訓(xùn)練構(gòu)建和最終分類器模型的形成；識別模塊包括測試圖片的輸入、車標識別和最終結(jié)果的輸出。

圖4 車標識別系統(tǒng)流程圖

2.2 網(wǎng)絡(luò)的構(gòu)建

為了提高識別的正確率，本文考慮綜合各個不同網(wǎng)絡(luò)模型，主要考慮ResNet50和Inception兩種模型，如圖5所示為本文構(gòu)建的算法模型。ResNet50和Inception兩種網(wǎng)絡(luò)訓(xùn)練時導(dǎo)出特征向量的時間較長，因而算法具體實現(xiàn)時將導(dǎo)出的特征向量先保存下來，以便后續(xù)的識別，減少計算量。

圖5 本文算法模型示意圖

2.3 實驗與結(jié)果分析

本文采用基于Keras的深度學(xué)習(xí)框架搭建車標識別網(wǎng)絡(luò)。Keras是一個高層神經(jīng)網(wǎng)絡(luò)API，Keras由純Python編寫而成。Keras的底層庫使用Theano或TensorFlow，這兩個庫也稱為Keras的反端，默認使用Theano作用后端來進行張量操作。車標圖片來自于交通監(jiān)控拍攝、網(wǎng)上以及線下采集的車輛圖片，共有10種車標圖片合計1 078張，其中訓(xùn)練集700張，測試集378張。圖6是部分車標實例，分別為奧迪、寶馬、奔馳、大眾、東風(fēng)、豐田、福特、奇瑞、現(xiàn)代和雪佛蘭。

圖6 10類車標圖片實例

首先，用本文算法對車標測試集進行了測試，識別結(jié)果見表1所示。由表1可以看出，10種車標整體的識別率達到了98.1%。對奧迪、寶馬、奔馳、福特等幾種常見的車標識別效果較好。

將本文算法與參考文獻[5]的方法進行實驗對比，即將本文和文獻[5]共有的車別類型進行測試，測試結(jié)果見表2，從表中數(shù)據(jù)可以看出，本文對車標識別的正確率有了一定的提高。實驗中，大眾和東風(fēng)車標的識別正確率較低，主要是因為測試集中較多大眾和東風(fēng)的車標圖片大小較小且圖片本身的分辨率較低，對識別結(jié)果產(chǎn)生了一定的影響。

表1 車標識別測試結(jié)果

表2 本文算法與文獻[5]的實驗對比結(jié)果

3 總結(jié)

本文提出了一種將ResNet50和Inception V3兩模型相結(jié)合的深度學(xué)習(xí)網(wǎng)絡(luò)。與傳統(tǒng)的卷積神經(jīng)網(wǎng)絡(luò)相比，由多個模型導(dǎo)出的特征向量相結(jié)合進行分類器的設(shè)計，提高了車標識別的正確率。

[1] Psyllos AP，Kayafas E.Vehicle logo recognition using a SIFT-based enhanced matching scheme[J].IEEE Transactions on Intelligent Transportation Systems，2010，11(2)：322-328.

[2] LLORCA D F， ARROYO R， SOTELO M A.Vehicle logo recognition in traffic images using hog features and SVM [C]//Proceedings of the 2013 16th International IEEE Conference on Intelligent Transportation Systems.Piscataway.NJ：IEEE， 2013：2229-2234.

[3] SUN Q， LU X， CHEN L， et al.An improved vehicle logo recognition method for road surveillance images [C] //Proceedings of the 2014 Seventh International Symposium on Computational Intelligence and Design.Piscataway.NJ：IEEE， 2014：373-376.

[4] SAM K-T， TIAN X-L.Vehicle logo recognition using modest AdaBoost and radial tchebichef moments [C] / / ICMLC 2012：Proceedings of the 2012 14th International Conference on Machine Learning and Computing.Singapore：IACSIT Press， 2012：91-95.

[5] 彭博，臧笛.基于深度學(xué)習(xí)的車標識別方法研究[J].計算機科學(xué)，2015，42(4)：268-273.

[6] Hinton G，Salakhutdinov R.Reducing the dimensionality of data with neural networks[J].Science，2016，313(5786)：504-507.

[7] Hinton G，Osindero S，Teh Y.A fast learning algorithm for deep belief nets[J].Neural Computation，2006，18：1527-1554.

[8] LeCun Y，Bottou L，Bengio Y，et al.Gradient-based learning applied to document recognition[J].Proceedings of the IEEE，1998，86(11)：2278-2324.

[9] He K，Zhang X，Ren S，et al.Deep residual learning for image recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Las Vegas:IEEE，2016：770-778.

[10] Szegedy C，Vanhoucke V，Ioffe S，et al.Rethinking the inception architecture for computer vision[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Las Vegas:IEEE，2016：2818-2826.

The Study and Implement of Vehicle Logo Recognition Algorithm Based on Deep Learning

CHEN Xiao-e，et al.

(DepartmentofInformationEngineering，F(xiàn)ujianChuanzhengCommunicationsCollege，F(xiàn)uzhou350007，China)

According to the algorithm of deep learning，the characteristic study and the characteristics of recognition can be operated by themselves.An algorithm of vehicle logo recognition based on Keras deep learning frame is proposed.This algorithm uses the combination of ResNet50 and Inception V3 models to make the training network to establish the classifier to recognize the vehicle logo images.The experiment result show that compared with the traditional convolution neural network classification method，it increases the recognition accuracy.

deep learning；Keras frame；vehicle logo recognition；feature extraction

2017-05-22

福建省中青年教師教育科研項目(JAT160704)

陳小娥(1984-)，女(漢)，福建永春，碩士 主要研究圖像處理。

10.3969/j.issn.1009-8984.2017.02.027

TP391.41

A

1009-8984(2017)02-0117-04