馬龍 劉勝
摘 要: 為了提高前方車輛檢測的準(zhǔn)確率和效率,提出了一種改進的多分辨率下的多特征提取的方向梯度直方圖(HOG)特征融合算法。首先將樣本擴縮為分辨率不同的圖像,后轉(zhuǎn)為YUV色彩空間,其次加權(quán)融合Y,U,V多通道方向梯度直方圖(HOG)形成訓(xùn)練特征,最后采用支持向量機(SVM)對融合后的訓(xùn)練特征車輛分類器訓(xùn)練和檢測。實驗表明,該算法比傳統(tǒng)HOG提取特征算法車輛檢測率更高,效率高達(dá)98.92%,并且在不同天氣狀況下均有良好的檢測效果和魯棒性。
關(guān)鍵詞: 方向梯度直方圖; YUV色彩空間; 支持向量機; 車輛檢測; 多特征融合
中圖分類號:TP391.4 文獻標(biāo)志碼:A 文章編號:1006-8228(2018)10-42-03
Abstract: To improve the accuracy and efficiency of vehicle detection, a multi-feature extractive histogram of oriented gradient (HOG) fusion algorithm is proposed. Firstly, the sample pattern is expanded and narrowed to images with different resolutions and converted to YUV color space, followed by weighted fusion Y, U, V multi-channel histogram of oriented gradient (HOG) training features, Finally, support vector machine (SVM) is used to train and detect the trained features. The experiment shows that this algorithm has a higher detection rate than the traditional HOG extraction feature algorithm, and the efficiency is as high as 98.92%,and have good detection effect and robustness under different weather conditions.
Key words: HOG; YUV color space; support vector machine; vehicle detection; multi-feature fusion
0 引言
快速且精準(zhǔn)的車輛檢測系統(tǒng)能在交通出行中避免潛在的危險,可以有效減少其帶來的交通隱患,其中基于視覺的車輛檢測也一直是圖像處理、計算機視覺的重要研究課題[1-2]。目前基于視覺的車輛檢測可以分為兩個階段:在特征提取階段,系統(tǒng)提取車輛的陰影[3],顏色[4],紋理等外觀特征,從而得到可能存在車輛的候選區(qū)域,以提高感興趣區(qū)域內(nèi)的檢測效率。檢測驗證階段主要有基于模板[5]和基于學(xué)習(xí)[6]兩種方法。基于模板[7]的方法魯棒性好,但實時性較差,檢測效果不穩(wěn)定;基于學(xué)習(xí)的方法是目前車輛檢測領(lǐng)域研究的重點,且檢測效果較好。本文在特征提取階段選用的是方向梯度直方圖(HOG), HOG具有注重輪廓不過多考慮輪廓內(nèi)部的形態(tài)特點和較強的描述物體輪廓能力。而在檢測驗證階段,考慮到車輛檢測本質(zhì)就是二分類的問題,支持向量機(SVM)本身具有較強的分類的能力,故選為本文車輛檢測的分類器。
1 車輛檢測方法概述
在傳統(tǒng)的HOG特征提取中,由于忽略了樣本圖像的色彩信息,當(dāng)車輛的灰度空間和地面很接近時,灰度域的局部梯度就不明顯,從而導(dǎo)致車輛檢測率降低和誤檢率增高,本文針對上述問題,提出了一種改進的多分辨率下的YUV色彩空間相融合的方向梯度直方圖(HOG)和支持向量機(SVM)的前向車輛檢測算法,具體步驟為:①將樣本分辨率擴縮為72×72,32×32之后,由RGB色彩空間轉(zhuǎn)為YUV色彩空間,分別提取圖像的Y、U、V三通道數(shù)據(jù)模板,保存為二維矩陣MU,MU,MV;②在對應(yīng)的二維矩陣中,分別提取和計算出多通道特征HY、HU、HV;③把得到HOG多通道特征進行加權(quán)融合成最終訓(xùn)練特征H_HOG;④融合后的H_HOG特征用支持向量機(SVM)進行訓(xùn)練,生成訓(xùn)練模型。
2 特征提取
2.1 RGB色彩空間轉(zhuǎn)為YUV色彩空間
YUV是一種基于人眼感知的色彩空間,其中“Y”表示明亮度,也就是灰階值;而“U”和“V”表示的則是色度,其作用是描述圖像的色彩和飽和度。采用YUV色彩空間是因為它的亮度信號Y和色度信號U、V是分離的,其中RGB色彩空間轉(zhuǎn)為YUV色彩空間公式為:
2.2 HOG特征提取與融合
方向梯度直方圖(histogram of oriented gradient, HOG)特征是用于計算機視覺和圖像處理中的對象檢測的特征描述器,已經(jīng)廣泛應(yīng)用于在行人檢測[8]和交通標(biāo)志檢測[9]中。以下是HOG特征提取具體的實現(xiàn)方法:
第一步:計算圖像像素點的梯度幅值與梯度方向:求出像素點橫坐標(biāo)和縱坐標(biāo)方向的梯度,據(jù)此計算每個像素(x,y)位置的梯度幅值和梯度方向。
第二步:將梯度方向均勻量化成9個方向(如圖1所示),每個像素采用梯度幅值加權(quán)的方式進行投票,計算出計算各方向梯度積分圖。
第三步:首先將圖像分成小的連通區(qū)域,稱之為細(xì)胞單元,然后獲取單元中每個像素的梯度或邊緣方向直方圖。最后組合這些直方圖形成特征描述器。為了提高性能,我們還可以把這些局部直方圖在圖像更大的范圍內(nèi)(我們把它稱為塊)進行對比度歸一化,圖2為本文的檢測窗口。
2.3 融合多通道HOG特征
通過上述(2.2)得到多通道特征HY,HU,HV,采用加權(quán)的方式進行特征融合,從而得到色彩圖像的HOG融合特征H_HOG,即為:
式⑶中,WY,WU,WV分別為HY,HU,HV的權(quán)重系數(shù),其值為,權(quán)重系數(shù)和為1,H_HOG為色彩圖像的HOG融合特征。按照上述方法提取正、負(fù)訓(xùn)練樣本的HOG特征后,采用支持向量機(SVM)進行分類器訓(xùn)練與分類。
3 SVM多特征融合訓(xùn)練分類法
3.1 車輛分類器訓(xùn)練
支持向量機(SVM)是通過將輸入向量映射到一個高維的特征空間,在該空間中構(gòu)造最優(yōu)分類面,從而實現(xiàn)二類或多類的分類。為了滿足HOG特征和檢測速度的需求,本文選用線性SVM二分類,進行HOG特征的訓(xùn)練分類。對于線性二分類,為了使得所有的訓(xùn)練樣本都能被正確地分類,轉(zhuǎn)化為一個二次型優(yōu)化問題:,則約束條件為:
3.2 支持向量機(SVM)參數(shù)優(yōu)選
選取合適的參數(shù)可以得SVM具有較強的分類能力和泛化能力,當(dāng)參數(shù)C越小時,分類器對樣本數(shù)據(jù)中誤判的樣本懲罰程度就越小,訓(xùn)練誤差則變大;當(dāng)參數(shù)C越大時,表示對錯誤例懲罰程度越大,可能會導(dǎo)致模型過擬合的情況。而核參數(shù)σ主要影響樣本數(shù)據(jù)在高維特征空間中分布的復(fù)雜程度。本文采用的是文獻[10]中在 LibSVM工具箱中,通過網(wǎng)格搜索和交叉驗證的方法得到的最優(yōu)參數(shù),其驗證精度為95.89%。
4 實驗結(jié)果與分析
本文的實驗程序是在python3.6下實現(xiàn)的,性能測試實驗在Inter(R) core i5 CPU、2G內(nèi)存,64位Windows 10操作系統(tǒng)的計算機上運行。其中訓(xùn)練和測試的正負(fù)樣本均使用GTI數(shù)據(jù)庫中的圖片,各800張,大小都是,其中正樣本包括顏色不同的越野車、轎車、貨車等;負(fù)樣本為包含路邊樹木、道路路面、護欄等背景圖。
為了證明本文算法的有效性和可行性,實驗從兩個部分驗證:
⑴ 將本文算法與傳統(tǒng)HOG檢測車輛算法進行對比實驗,分別統(tǒng)計得到車輛檢測的結(jié)果如表1,從表可知本文算法在檢測率和誤檢率兩項指標(biāo)均優(yōu)于傳統(tǒng)算法。
⑵ 測試選自不同天氣情況下的車輛前置攝像頭的真實數(shù)據(jù),結(jié)果如圖3所示(分別為晴天、霧天、雨天),均取得良好的檢測效果。
5 結(jié)束語
針對實際應(yīng)用中對于車輛檢測的檢測效率和準(zhǔn)確率,本文在提取的梯度輪廓特征的基礎(chǔ)上加以改進,在多分辨率下提取融合的YUV色彩空間的HOG特征,比傳統(tǒng)的HOG特征包含更多色彩信息,能有效避免傳統(tǒng)提取算法干擾物對檢測效果的影響。經(jīng)過多次實驗,證明了檢測的準(zhǔn)確率有明顯提高,能適應(yīng)不同天氣的條件且具有較好的魯棒性,在總體性能上有較大的提升。但本文算法對前方較多車輛和重疊嚴(yán)重的情況下檢測不太理想,這將作為后續(xù)的研究重點,進而提高車輛檢測系統(tǒng)的性能。
參考文獻(References):
[1] Li Xing,Guo Xiao song. A HOG feature and SVM based method for forward vehicle detection with single camera [C]//IEEE International Conference on Intelligent Human -Machine Systems and Cybernetics,2013:263-266
[2] 楊強,齊美彬,蔣建國.一種用于車輛檢測的選擇性背景更新方法[J].合肥工業(yè)大學(xué)學(xué)報(自然科學(xué)版),2011.34(4):509-514
[3] Han S, Han Y, Hahn H. vehicle detection Method using Haar like Feature on Real Time System[J].Engineer and Technology,2009.59:455-459
[4] Guo D,F(xiàn)raichard T,Xie M,et al. Color Modeling by Spherical Influence Field in Sensing Driving Environment[C]//Proceedings IEEE Intelligent Vehicle Symposium.2000:249-254
[5] Handmann U,Kalinke T,Tzomakas C,et al. An Image Processing System for Driver Assistance [J]. Image and Vision Computing,2000.18(5):367-376
[6] Wu Jun-wen,Zhang Xue-gong. A PCA Classifier and Its Application in Vehicle Detection[C]//Proceeding IEEE International Joint Conference on Neural Networks,2001:600-604
[7] Gepperth ART. Co training of context models for real timevehicle detection[C]//2012 IEEE Intelligent Vehicles Symposium,2012:814-820
[8] Zhu Qiang,Yeh Meichen, Cheng K-T. Fast Human Detection Using a Cascade of Histograms of Oriented Gradients[C]//proceedings IEEE International Conference on Computer Vision & Pattern Recognition,2006.2:1491-1498
[9] 陳龍,潘志明,毛慶洲等.利用HOG LBP自適應(yīng)融合特征實現(xiàn)禁令交通標(biāo)志檢測[J].武漢大學(xué)學(xué)報(信息科學(xué)版),2013.38(2):191-194
[10] 李星,郭曉松,郭君斌.基于HOG特征和SVM的前向車輛識別方法[J].計算機科學(xué),2013.40(11):329-332