戚秀真,李娜
(1.長安大學信息工程學院,西安 710064;2.西安郵電大學通信與信息工程學院,西安 710121)
基于視頻的車流量統(tǒng)計算法
戚秀真1,李娜2
(1.長安大學信息工程學院,西安 710064;2.西安郵電大學通信與信息工程學院,西安 710121)
提出一種基于視頻的車流量統(tǒng)計算法,首先采用高斯混合模型建立背景模型,然后采用背景差分法提取前景,通過形態(tài)學處理得到運動車輛,接著利用多示例學習的方法對目標進行跟蹤。在Visual Studio 2008上結(jié)合OpenCV 2.4.4編程實現(xiàn)并驗證該算法,取得較好的效果。
計算機視覺;車流量統(tǒng)計;目標檢測
隨著圖像處理技術(shù)的飛速發(fā)展,視頻檢測逐漸體現(xiàn)出其優(yōu)越性:安裝維護方便,不破壞路面,不影響道路的正常使用;檢測范圍廣,能夠檢測出靜止車輛;系統(tǒng)采用模塊化、結(jié)構(gòu)化設(shè)計,擴展性好,運行效率高。因此,視頻檢測技術(shù)在交通監(jiān)控、交通管理等領(lǐng)域得到了廣泛的應(yīng)用[1-2]。
本文在研究國內(nèi)外相關(guān)技術(shù)的基礎(chǔ)上,提出了一種基于視頻的車流量統(tǒng)計算法,并在Visual Studio 2008上結(jié)合OpenCV 2.4.4編程實現(xiàn),取得了較好的效果。
基于視頻的車流量統(tǒng)計算法主要包括車輛檢測和車輛跟蹤兩個部分。首先,讀入視頻序列,利用高斯混合模型(Gaussian Mixture Model,GMM)對背景建模,然后利用背景差分法,提取前景圖像,并進行形態(tài)學處理,從而檢測得到運動目標。對目標所在區(qū)域提取Haar-like特征,訓練并更新多示例學習(Multiple Instance Learning,MIL)分類器,實現(xiàn)對目標的連續(xù)跟蹤。最后統(tǒng)計感興趣區(qū)域內(nèi)單位時間內(nèi)的車輛總數(shù),即得到車流量。
圖1 車流量統(tǒng)計流程
(1)車輛檢測
背景建模中較常用的是高斯混合模型,但實驗發(fā)現(xiàn):經(jīng)GMM建模后,利用背景差分法檢測得到的前景圖像中,車輛內(nèi)部容易出現(xiàn)空洞,這給后續(xù)車輛的檢測和跟蹤帶來了困難。本文采用GMM結(jié)合背景差分法得到二值化圖像,然后在目標像素的鄰域范圍內(nèi),將與目標像素具有相似鄰域結(jié)構(gòu)的背景像素標記為目標像素,這將在一定程度上,填補車輛內(nèi)部空洞,便于對車輛區(qū)域進行進一步處理。具體方法如下:
對于一幅M×N的圖像I,對坐標為(x,y)的像素點求它的絕對坐標,記為j,即 j=(y-1)×M+x,因此,(x,y)處的像素點即為像素j。記Wjr表示以像素j為中心、大小為r×(rr為奇數(shù))的搜索窗,像素p是落在這個搜索窗內(nèi)的像素點。在搜索窗Wjr內(nèi),像素p與像素j之間的相似度djp由下式定義:
其中,h是控制djp衰減程度的參數(shù);v(Nj)表示以像素j為中心、大小為f×f的相似窗Nj上的灰度向量,為利用高斯核函數(shù)定義的像素j和像素p之間的距離,其中,α>0是高斯核函數(shù)的標準差;δj是歸一化常數(shù),具體定義為:
像素p與像素j之間的相似度djp反映了兩個像素之間鄰域結(jié)構(gòu)的相似性。如果djp越大,說明兩個像素之間鄰域結(jié)構(gòu)越相似;反之,兩個像素之間鄰域結(jié)構(gòu)越不相似。如果兩個像素的鄰域結(jié)構(gòu)很相似,而其中一個像素點為目標,那么另一個像素點也應(yīng)該是目標。
經(jīng)過這樣處理后,得到一個包含更多目標信息的二值化圖像,利用腐蝕和膨脹等形態(tài)學處理方法對該圖像進行處理,去掉離散的點,填充小的空洞,然后采用連通域標記方法,得到車輛的候選區(qū)域,把距離較近的連通域進行合并,把面積較小的連通域進行刪除,得到的就是候選目標區(qū)域。由于該區(qū)域中可能存在多個車輛粘連的情況,按照文獻[3]中的方法,利用占空比對粘連區(qū)域進行檢測,然后提取粘連區(qū)域的骨架,檢測骨架上的角點,根據(jù)角點的聚類結(jié)果對相互遮擋的車輛進行準確分割,最終得到的每一個連通域就代表著一輛車,由此完成運動車輛檢測。
(2)車輛跟蹤
本文采用MIL的方法[4]對運動目標進行跟蹤。在檢測出的運動車輛的鄰域內(nèi)提取正包和負包,通過對訓練包的學習,得到MIL分類器,然后讀入下一幀圖像,利用訓練得到的MIL分類器,在當前幀中找到分類器響應(yīng)最大的區(qū)域作為跟蹤結(jié)果,然后在其鄰域內(nèi)提取正包和負包,更新MIL分類器,在下一幀中判斷目標所在位置,如此重復(fù),直到目標消失。具體方案如下:
①提取正包和負包
提取正包和負包的方法為:在第t幀目標所在位置lt*的鄰域內(nèi),找到兩個圖像集合 Xγ={x:||l(x)-lt*||<γ}和 Xγ,β={x:γ<||l(x)-lt*||<β},將 Xγ作為正包,Xγ,β作為負包。其中,γ,β分別是正、負包的選擇半徑,l(x)為圖像塊x的位置,用目標中心點坐標(x,y)表示。
②訓練MIL分類器
第一步,訓練M個弱分類器。每一個弱分類器hk(x)由特征fk及其所服從分布的參數(shù)唯一確定。假設(shè)p(y=1)=p(y=0),可計算得到弱分類器hk(x),即:
經(jīng)訓練最終得到M個弱分類器φ={h1,h2,...,hM}。
第二步,構(gòu)造MIL分類器。從M個弱分類器中依次選擇K個弱分類器構(gòu)成強分類器,使得包的對數(shù)似然函數(shù)最大。最終得到的分類器為:
③判斷目標在當前幀中的位置
在當前幀中,根據(jù)前一幀中目標位置lt-1*,在其鄰域內(nèi)提取圖像集合,利用MIL分類器計算出與目標相似性最大的圖像塊,并把它的位置作為當前幀中目標所在位置lt*。
④更新MIL分類器
第一步,更新特征分布的參數(shù)。分別假設(shè)正包和負包中示例的特征均服從高斯混合模型和單高斯分布,分別更新其均值和方差。
以負包中示例為例,其特征服從高斯分布:
按下式更新其參數(shù):
其中,μ0,t,σ0,t2為t時刻高斯分布的均值和方差,為t時刻負包中所有示例第k個特征的均值,為t時刻負包中所有示例第k個特征的方差,0<ζ<1為參數(shù)更新速率,通過目標與模型的相似度來定義。
第二步,重新訓練M個弱分類器。
第三步,重新構(gòu)造MIL分類器。
本文在Visual Studio 2008平臺上,利用機器視覺庫OpenCV 2.4.4,編程實現(xiàn)了該算法,如圖2所示。首先設(shè)定用戶感興趣的區(qū)域,算法自動對感興趣區(qū)域內(nèi)的車輛進行檢測跟蹤和計數(shù)。對1小時視頻內(nèi)的交通量進行統(tǒng)計,準確率可達95%。分析漏統(tǒng)計和錯統(tǒng)計的原因主要是車輛之間嚴重遮擋,以及光線對背景建模的影響所致。因此,下一步可從提高背景建模準確率和解決遮擋情況下的跟蹤入手,提高交通量統(tǒng)計的準確性。
本文提出了一種基于視頻的車流量統(tǒng)計算法,采用GMM對感興趣區(qū)域的背景建模,然后利用背景差分法提取目標,利用MIL方法對其進行跟蹤,并在Visual Studio 2008上結(jié)合OpenCV編程實現(xiàn)并驗證了該算法。但是當目標相互遮擋和光線變化時,算法性能有所下降,需要進一步深入研究。
圖2 軟件界面
[1]丁兆威.智能交通發(fā)展前景[J].中國公共安全(綜合版),2008(3):5-7.
[2]陳艷,何春明,智能交通系統(tǒng)應(yīng)用現(xiàn)狀及其存在問題分析[J].交通標準化,2007(8):62-65.
[3]李娜,陳俊艷,劉穎,等.基于骨架角點檢測的粘連車輛分割[J].西安郵電大學學報,2015,20(6):14-18.
[4]李娜,趙祥模,趙鳳,等.基于外觀模型的目標跟蹤算法研究進展[J].計算機工程與科學,2017,39(3):524-533.
Traffic Flow Statistics Algorithm Based on Video
Qi Xiu-zhen1,LI Na2
(1.School of Information Engineering,Chang'an University,Xi'an 710064;2.School of Telecommunications and Information Engineering,Xi'an University of Posts and Telecommunications,Xi'an 710121)
Presents a traffic flow statistics algorithm based on video,firstly,the background model is established by Gaussian mixture model,and then the background subtraction method is used to extract the foreground.After the morphological processing,the moving vehicles are detected.And multiple instance learning is adopted for object tracking.The algorithm is implemented in Visual Studio 2008 combined with OpenCV 2.4.4,and provides good performance.
Computer Vision;Traffic Flow Statistics;Object Detection
陜西省西安市碑林區(qū)科技項目(No.GX1502)
1007-1423(2017)32-0079-03
10.3969/j.issn.1007-1423.2017.32.018
戚秀真(1981-),女,安徽碭山人,博士,研究方向為智能交通、圖像處理
李娜(1981-),女,四川瀘州人,碩士,研究方向為圖像處理、模式識別
2017-08-15
2017-10-28