基于SSD的道路車輛目標檢測

張向清 楊銳 白濤 王金妮

摘?要：針對道路交通場景下的3類目標（小車，公交車，卡車），本文先創建特定場景下的目標數據集，并對其中的數據標注規則，標注方法以及標注手段進行實驗。采用SSD目標檢測算法，訓練得到特定場景下的目標檢測模型，并融合數據增廣等方式對數據集優化。通過實驗分析，本文的訓練模型檢測準確度高，并且模型的魯棒性好。

關鍵詞：SSD;自制數據集;檢測模型

道路交通運輸行業在國民經濟發展中發揮著重要作用，越來越多的機動車涌入道路，如果對道路中的目標不加以交通管制或采取一定的措施防范控制道路中目標的數量，將會引發更多的交通問題，例如：交通擁堵和交通違章越來越嚴重，致使交通中文明出行變得微乎其微。隨著互聯網時代的到來，各種智能硬件設備應用到交通場景，有效的緩解了部分交通問題。但是，由于硬件設備的安裝繁雜、老化嚴重以及更新成本巨大，使交通管理部門的運營成本昂貴，本文分析一種基于視頻分析的交通監控算法，不僅降低了交通管理的運營成本，而且其準確度相比部分硬件設備有了明顯的提高。

1 常見的目標檢測算法

目標檢測目前有one-stage和two-stage兩種，two-stage指的是檢測算法需要分兩步完成，首先需要獲取候選區域，然后進行分類，比如R-CNN系列;與之相對的是one-stage檢測，可以理解為一步到位，不需要單獨尋找候選區域，典型的有SSD/YOLO。

1.1 基于候選區域的目標檢測

首先將每個像素作為一組，求出各自的特征，然后將兩個最為接近的特征進行組合。為了避免大目標對小目標的淹沒，需設置不同尺度大小的ROI區域，在不同的卷積層進行矩形框的組合。這些區域被轉換為固定大小的圖像，并分別送到卷積神經網絡中。在輸出層采用SVM分類器，使用Logic函數計算損失來修正邊界框，以實現目標分類并得到邊界框。

1.2 基于端對端學習的目標檢測

該類方法無需預先提取候選區域，以YOLO目標檢測算法為例，整個模型的訓練和檢測都進行了簡化，其將目標檢測的類別和位置同時進行預測，可以歸納為回歸問題。主要步驟為：卷積層進行圖像特征提取;全連接層用于后續類別和目標位置的回歸預測。

2 SSD目標檢測算法

2.1 算法原理

一種基于回歸的SSD檢測模型，其不同于候選框提取的目標檢測算法，例如：Faster R-CNN檢測算法，此算法不再提取候選框以及對像素或特征進行采樣，其將所有計算封裝在單網絡中使得SSD容易訓練或直接集成到需要檢測組件的系統。同時，SSD對不同大小的輸入圖像的目標檢測具有更好的精度，更快的速度。

SSD的模型結構使用VGG-16網絡作為基礎，然后添加額外的特征提取層，在不斷進行卷積的時候，得到的特征圖尺寸逐漸減小而后輸入到Extra feature層用來得到多個尺度檢測的預測值，形成了多個輸出路徑。為了簡化說明，我們把模型簡化為單輸出的網絡，即只有一個輸出路徑，每個路徑都是一個全卷積網絡。

2.2 模型的訓練

SSD算法中實現了多種數據集的訓練方法，而對于自己的數據集需要修改文件中的不同參數設置以及對訓練方法進行改進才能使得訓練出來的模型適合特定的應用場景。下面將對訓練方法做具體分析：

（1）采集數據樣本，對分辨率不高或目標不明顯的圖像和視頻幀刪除，統計所有樣本，并把目標的矩形坐標保存到文本文件，保證圖像和目標矩形框能一一對應。

（2）生成訓練數據集，先將圖片和標簽文件轉化為LevelDB或lmDB數據庫，我們將訓練驗證集和測試集以一定的比例劃分開，最后將這些數據庫格式的文件輸入到深度學習的網絡中。

3 實驗結果

通過對不同數據集的訓練，得到其不同輸出結果的訓練模型。并將訓練模型用于檢測高速公路場景內的目標，實驗發現：小車類的檢測準確度較高其平均準確度為93.2%，公交類的檢測平均準確度為89.7%，卡車類的準確度為87.4%，主要由于卡車類的訓練目標較少。可以看出，這三類目標檢測性能較好，基本符合應用場景。

其次，我們對SSD中不同的基礎網絡模型也進行了測試分析，可以發現：當使用ResNet101基礎網絡框架再結合分類器能夠檢測出場景內的更多目標，而基于VGG19和ZFNet5基礎網絡結構檢測三類目標的準確度略低，說明在一定范圍內卷積層數越多，檢測準確度越好。另外，對不同輸入尺度的圖像或視頻流（本文分析了300x300和512x512），模型的檢測速度差異很大，對于場景中目標不多且明顯的車輛，可以采用速度較快的300x300的圖像大小作為輸入。

4 結語

本文創建了特定場景下的數據集，訓練不同的網絡模型并對網絡訓練過程中的參數微調和模型優化進行測試分析。通過實驗分析，本文采用的SSD目標檢測算法，在自制數據集上性能表現良好，后期主要研究如何把模型參數進行壓縮，在嵌入式設備中進行實時檢測目標。

參考文獻：

[1]張向清.高速公路場景下基于深度學習的車輛目標檢測與應用研究[D].長安大學.

[2]張文桂.基于深度學習的車輛檢測方法研究[D].華南理工大學，2016.

[3]張向清.基于深度學習方法的復雜場景下車輛目標檢測[J].計算機應用研究，2018，35（4）：1270-1273.

作者簡介：張向清（1991-），男，陜西榆林人，碩士，信息與通信工程專業。