基于深度學習的車輛檢測系統設計

黎雄 陳雨聰

摘 ?要：針對目前智能交通系統實時道路車輛檢測中存在的檢測效率不高等問題，設計了一種基于深度學習的道路車輛檢測算法。首先在搭建好的平臺下，通過卷積神經網絡對采集的車輛圖像數據集進行訓練，得到訓練后的模型;其次，對該模型內部的層結構進行可視化;最后，通過調節各網絡參數及層結構對該模型進行優化。訓練的模型通過實驗測試，分別對圖片和視頻進行檢測，圖像識別準確率高，檢測速度快，跟蹤精度高，可應用于實時交通系統的檢測。

關鍵詞：深度學習;卷積神經網絡;車輛識別;OpenCV

中圖分類號：TP183 ? ? 文獻標識碼：A

Design of Vehicle Detection System based on Deep Learning

LI Xiong1， CHEN Yucong2

（1.Xi'an Institute of Electromechanical Information Technology， Xi'an 710065， China;

2.School of Computing， Xianyang Normal University， Xianyang 712000， China）

42340658@qq.com; 943858467@qq.com

Abstract： Aiming at the low detection efficiency of real-time road vehicle detection in intelligent transportations system， this paper proposes to design a road vehicle detection algorithm based on deep learning. Firstly， based on the built platform， convolutional neural network is used to train the collected vehicle image dataset to obtain the trained model. Secondly， the layer structure inside the model is visualized. Finally， the model is optimized by adjusting all of the network parameters and layer structure. The trained model has been tested by experiments to detect pictures and videos respectively. Experiment results show that the proposed model can be applied to the detection of real-time traffic systems with high image recognition accuracy， fast detection speed， and high tracking accuracy.

Keywords： deep learning; convolutional neural network; vehicle recognition; OpenCV

1 ? 引言（Introduction）

隨著大數據時代的到來，越來越多的數據需要處理。而人們對信息處理的要求也越來越嚴格，既要達到高準確率，又要達到較高的運行速率。卷積神經網絡是為了識別二維圖像而設計的多層感知器，在處理圖像方面具有良好的容錯能力與并行的處理以及學習能力。在深度學習圖像識別應用上，卷積神經網絡具有較高的準確率和處理速度，所以可以作為理論依據來研究動態車輛識別。對深度學習動態車輛的研究，主要是通過攝像頭實時監控車輛的位置，繼而來判斷車輛的類別，這一領域的研究對今后智能輔助領域的數據處理發展有著至關重要的作用[1-2]。與此同時，也面臨著艱巨的挑戰，如車輛檢測方面包括視頻序列中背景的更新、復雜環境下背景圖片的遮擋和運動車輛在道路中停止時對跟蹤的影響，以及復雜環境下車輛檢測的實時性要求等[3]。因此，設計一個能夠在交通環境下對各種車輛的各種特征進行高效實時的檢測算法勢在必行。

2 ? 檢測算法概述（Overview of detection algorithms）

目前，多目標檢測算法主要分為傳統方法和基于深度學習的方法。傳統方法主要有光流法[4]、卡爾曼濾波[5]等，傳統目標檢測算法建立在手工設計特征的基礎上，由于算法的實時性要求、圖像的表示與預處理技術等問題，對采集到的復雜的圖像特征無法高速計算與處理。其中，光流法利用圖像序列中像素在時間域上的變化以及相鄰幀之間的相關性，根據上一幀與當前幀之間的對應關系，計算得到相鄰幀之間物體的運動信息，在無遮擋、光線好的情況下檢測效果較好。但在實際應用中，采集的圖像都存在噪聲、陰影、遮擋、光照等干擾因素，所以光流法的抵抗性較差，檢測出的目標會出現內容缺失的情況[6-7]，需要借助其他算法。卡爾曼濾波（KCF）算法利用循環移位進行稠密采樣，利用快速傅里葉變換（FFT）進行分類器的訓練，同時結合了多通道的HOG特征。其大致的流程是，先利用循環矩陣不斷對圖像移位，得到多個樣本，在第t 幀中的當前位置附近利用這些樣本訓練一個分類器，該分類器可以對框中是否有人做出一個概率響應。因此，先用循環矩陣對前一幀的區域進行循環移位得到若干樣本，然后用前一幀訓練的分類器分類得到輸出響應，以響應最大的作為預測位置，然后再訓練、再預測。

近幾年來，目標檢測算法取得了很大的突破。比較流行的算法可以分為兩類，一類是Two-stage算法，基于候選區域（Region Proposal）的R-CNN系列算法（R-CNN、Fast R-CNN、Faster R-CNN），先使用啟發式方法（Selective Search）或者CNN網絡（RPN）產生Region Proposal，然后再在Region Proposal上做分類與回歸;另一類是One-stage算法，基于Yolo、SSD算法，這類方法則不需要產生候選框，僅僅使用一個CNN網絡直接預測不同目標的類別與位置。第一類算法準確度高，速度較慢;第二類算法速度快，準確性較低[8-12]。

本設計采用深度學習的卷積神經網絡LeNet-5模型的網絡框架建立模型，并進行訓練。卷積神經網絡能夠很好地利用圖像的結構信息。LeNet-5是一個較簡單的卷積神經網絡，該結構包括輸入層、卷積層、池化層和輸出層[13]。輸入層輸入的是二維圖像，該圖像經過兩次卷積層提取特征，經過卷積之后圖像的邊界變得更加直觀，然后送到池化層。為了提取最明顯的特征，通過最大池化法來降低參數，再經過連接層，連接所有的特征，最后將輸出值送給分類器，使用Softmax分類函數進行輸出。

3 ?深度學習模型結構（Structure of deep learning model）

3.1 ? 模型結構

該模型由TensorFlow 2.4.0搭建，使用的是Sequential順序模型，層與層之間只有相鄰關系，沒有跨層連接，編譯速度快，操作簡單。其模型結構如圖1所示。

輸入層是32×32尺寸的圖片，其中每一個像素點都可以看作一個神經元，通過神經元的傳遞來計算數據處理得到輸出圖像。輸入層到C1層經過卷積的操作，卷積核的大小是5×5，使得圖像從32×32變成28×28的圖像。C1層由八個特征圖組成，輸入層到C1層是一一對應關系。

C1層到S1層采用的是下采樣方式，使得28×28的圖片變成14×14的大小，如果使用下采樣方式就能夠更少地利用參數，使得圖像符合顯示區域的大小，同時還可以保留有效圖片數據。S1中的神經元大小與C1層2×2的大小一一對應，通過計算得到S1層的神經元。同理可知，其他的卷積和下采樣方式一樣，只是處理的數據參數不一樣。

由神經元的概念可知，一個輸出值需要知道權值和偏置的參數大小，而LeNet-5采用的是32×32的二維神經元結構網，可通過表格的形式來顯示為何選用卷積神經網絡而不采用直接神經元處理方式，如表1所示。

由表1可知，使用卷積神經網絡能夠使參數大大地減小，因為卷積神經網絡具有權值共享和參數減小的優點。S1層與C3層使用的不是全連接形式，而是局部連接。其他步驟均使用了全連接方式，如表2所示是S1層與C3層的連接表。

3.2 ? 模型訓練

先收集相關車輛的數據集，然后讀取數據集，將數據集進行歸一化處理，統一圖片大小，將每張圖片所處的文件名作為該圖片的標簽，然后把所有圖片和其標簽打包在一起，供訓練模型使用。將數據集分成兩部分，一部分作為訓練集，一部分作為測試集。

采用Adam（Adaptive Moment Estimation）這個優化算法來快速擬合高緯度的數據，損失函數為sparse_categorical_crossentropy，其目標是數字編碼。使用之前建立好的模型進行30 個輪次訓練，然后保存訓練結果。模型訓練日志結果如圖2所示，訓練流程圖如圖3所示。

4 ?車輛檢測系統的實現（Realization of vehicle detection system）

4.1 ? 在VS下OpenCV的配置

首先，在VS2017中點擊文件→新建→項目→控制臺應用→確定，創建好工程。其次，點擊resource files→添加→新建項→添加C++文件。第三，項目配置屬性。在視圖上點擊屬性管理器，設置系統為64 位。找到Microsoft.Cpp.X64.user并且雙擊“配置屬性”，在VC++目錄中找到包含目錄與庫目錄進行配置，如圖4所示。之后在鏈接器的輸入中找到附加依賴項進行配置，如圖5所示。至此，完成了OpenCV在VS下的配置。

4.2 ? 在VS下Qt的配置

打開VS2017，點擊文件→新建→項目，在Visual C++下找到Qt，選擇Qt Widgets Application，新建名為vehilce_detect工程。在項目的屬性管理器中選擇X64的屬性并且配置VC++目錄的包含目錄和庫目錄，與配置OpenCV相同。然后，進行Qt圖形用戶界面的設計，設計效果如圖6所示。

4.3 ? 系統測試

系統運行后，單擊“選擇圖像”按鈕，在彈出的界面中選擇要檢測的圖像，圖像支持JPEG、BMP、PNG格式。選擇完成后，所選擇的圖像會呈現在檢測界面上，之后點擊“識別”按鈕，打開的圖像如圖7所示。

單擊“視頻檢測”按鈕，在彈出的界面中選擇要檢測的視頻，支持avi、MP4等格式，選擇完成后，會在視頻窗口顯示實時視頻車輛檢測，如圖8所示。

5 ? 結論（Conclusion）

此次采用深度學習中的卷積神經網絡LeNet-5模型的網絡框架對車輛數據集進行訓練，通過在VS中搭建OpenCV與Qt，建立檢測系統，并加載訓練模型進行測試。實驗首先驗證算法模型的準確性，加載訓練的模型，該系統能夠準確地檢測圖片中的車輛，并能實時檢測與追蹤視頻中的車輛。同時，驗證了開發的檢測系統的有效性，為后續模型檢測及跟蹤算法驗證提供了方便。

參考文獻（References）

[1] 袁氫.基于特征融合與神經網絡的深度學習圖像識別技術研究[D].武漢：武漢科技大學，2020.

[2] 馮霞，杜佳浩，段儀濃，等.基于深度學習的行人重識別研究綜述[J].計算機應用研究，2020，37（11）：3220-3226，3240.

[3] 何文偉.交通環境下的車輛檢測與跟蹤算法研究與實現[D].成都：電子科技大學，2015.

[4] 尹梓睿，張索非，張磊，等.適于行人重識別的二分支EfficientNet網絡設計[J].信號處理，2020，36（09）：1481-1488.

[5] 趙瑜，方亮，周云，等.基于深度學習理論和壓縮感知技術的車輛識別與跟蹤研究[J].地震工程與工程振動，2021，41（05）：

206-214.

[6] 李舒航，高旭，劉兆武，等.基于灰度插值的條紋圖像位移亞像素檢測算法[J].光學學報，2021，41（10）：125-133.

[7] 張可，楊燦坤，周春平，等.無人機視頻圖像運動目標檢測算法綜述[J].液晶與顯示，2019，34（1）：98-109.

[8] ZHOU L， ZHOU C C， CHEN S B， et al. A sparsetransformer-

based approach for image captioning[J]. IEEE Access， 2020， 99（8）：213437-213446.

[9] 梅瑩.基于卷積神經網絡的隧道行人目標檢測算法研究[D].重慶：重慶大學，2019.

[10] WAN L J， CHEN Y W， LI H Y， et al. Rolling-element bearing fault diagnosis using improved LeNet-5 network[J]. Sensors （Basel， Switzerland）， 2020， 20（6）：1693-1711.

[11] 王宇歌，張涌，黃林雄，等.基于卷積神經網絡的麥穗目標檢測算法研究[J].軟件工程，2021，24（8）：6-10.

[12] ZHONG Z Y， SUN L， HUO Q. An anchor-free

region proposal network for faster R-CNN-based text detection approaches[J]. International Journal on Document Analysis and Recognition （IJDAR）， 2019， 22（3）：

315-327.

[13] 王潤正，高見，黃淑華，等.基于知識蒸餾的惡意代碼家族檢測方法[J].計算機科學，2021，48（1）：280-286.

作者簡介：

黎 ? 雄（1980-），男，碩士，高級工程師.研究領域：信號與圖像處理.本文通信作者.

陳雨聰（1998-），男，本科，助理工程師，研究領域：計算機應用.