顧晨亮，楊 恒，劉友波，張 晗，張 勁，何 凌

(四川大學(xué) 電氣工程學(xué)院，四川 成都 610065)

0 引言

隨著近年來(lái)我國(guó)基礎(chǔ)建設(shè)水平的飛速發(fā)展，建設(shè)項(xiàng)目的規(guī)模和復(fù)雜性加大，參與方增多，工程項(xiàng)目的風(fēng)險(xiǎn)及管理難度也不斷加大[1]。在施工過(guò)程中，工程車輛的不規(guī)范活動(dòng)可能會(huì)造成安全距離縮短，容易引起事故，從而帶來(lái)巨大的經(jīng)濟(jì)損失或人員傷亡，因此工程車輛的檢測(cè)識(shí)別是建設(shè)施工項(xiàng)目巡檢故障識(shí)別中的重要研究?jī)?nèi)容，具有重要的理論價(jià)值和工程應(yīng)用價(jià)值[2]。

傳統(tǒng)方法中對(duì)于工程車輛檢測(cè)的研究，大多集中在智能監(jiān)控領(lǐng)域，通過(guò)固定攝像機(jī)進(jìn)行圖像或視頻的采集，使用背景差分法、統(tǒng)計(jì)模式法等算法來(lái)進(jìn)行工程車輛的檢測(cè)[3-5]。以上方法中對(duì)于施工車輛的大小、角度、背景等都較為固定，具有一定的應(yīng)用局限性，泛化能力較差。

近些年，隨著深度學(xué)習(xí)在計(jì)算機(jī)視覺(jué)領(lǐng)域中迅速的發(fā)展，已有部分學(xué)者使用基于卷積神經(jīng)網(wǎng)絡(luò)的檢測(cè)算法對(duì)工程車輛檢測(cè)方面進(jìn)行了探索，王世凱[6]從卷積核尺寸設(shè)計(jì)、分類器設(shè)計(jì)等方面對(duì)Faster R-CNN[7]進(jìn)行改進(jìn)，提出了TN-CNN來(lái)對(duì)工程車輛進(jìn)行檢測(cè)與識(shí)別；李千登等[8]使用Faster R-CNN結(jié)合攝影參數(shù)與報(bào)警規(guī)則進(jìn)行了叉車危險(xiǎn)操作行為的檢測(cè)。但上述研究中均未考慮工程車輛的遮擋問(wèn)題，而馮小雨等[9]針對(duì)遮擋檢測(cè)效果差的問(wèn)題，提出了局部標(biāo)注和時(shí)空上下文策略，一定程度上降低了遮擋對(duì)檢測(cè)的影響。

盡管目前已有針對(duì)于車型識(shí)別的方法，但多是基于交通與停車場(chǎng)等監(jiān)控覆蓋較為全面的場(chǎng)景[10-12]，由于施工工地地形復(fù)雜、工程車輛特征相似性較大以及工程車輛相互遮擋等原因，工程車輛檢測(cè)識(shí)別的難度較大，相關(guān)研究仍比較少。因此，本文提出基于自適應(yīng)局部斥力與歸一化面積損失的工程車輛目標(biāo)檢測(cè)算法，針對(duì)遮擋問(wèn)題與特征相似性進(jìn)行優(yōu)化，并結(jié)合聚類算法設(shè)定初始錨框減少預(yù)測(cè)框的偏移，從而實(shí)現(xiàn)對(duì)壓路機(jī)、挖掘機(jī)、裝載機(jī)3類工程車輛的檢測(cè)與識(shí)別，算法可有助于在復(fù)雜、遮擋下的施工場(chǎng)景下對(duì)工程車輛進(jìn)行有效的安全監(jiān)管。

1 工程車輛檢測(cè)與識(shí)別算法框架

本文算法采用無(wú)人機(jī)采集航拍視頻序列為數(shù)據(jù)集，使用自適應(yīng)局部斥力損失與歸一化面積損失，并結(jié)合聚類算法設(shè)定初始錨框減少預(yù)測(cè)框的偏移。以Faster R-CNN[7]作為基準(zhǔn)檢測(cè)網(wǎng)絡(luò)為例，使用本文提出的多種損失函數(shù)與優(yōu)化算法對(duì)工程車輛在圖像中進(jìn)行3類工程車輛目標(biāo)區(qū)域標(biāo)定，實(shí)現(xiàn)對(duì)壓路機(jī)、挖掘機(jī)、裝載機(jī)3類工程車輛的高精度與強(qiáng)魯棒性的自動(dòng)檢測(cè)與識(shí)別，算法流程如圖1所示。

圖1 工程車輛目標(biāo)檢測(cè)系統(tǒng)的算法流程Fig.1 Algorithm procedure of target detection system for engineering vehicles

2 自適應(yīng)局部斥力損失與歸一化面積損失

在施工場(chǎng)景下，由于環(huán)境的復(fù)雜以及車輛的不斷運(yùn)動(dòng)，再加上車輛的形態(tài)變化與形狀的相似性，導(dǎo)致檢測(cè)過(guò)程中存在車輛之間的遮擋情況與車輛自身局部特征與全局特征相似的情況。因此，本文針對(duì)這2個(gè)問(wèn)題提出自適應(yīng)局部斥力損失(Self-adaptive Local Exclusion Loss，SCLEL)與歸一化面積損失(Normalized Area Loss，NAL)，全局損失函數(shù)表達(dá)式如式(1)所示：

L=Lbase+α*LSCLEL+β*LNAL

(1)

式中：Lbase為基礎(chǔ)檢測(cè)網(wǎng)絡(luò)的損失函數(shù)，如Faster R-CNN中使用的smoothL1損失函數(shù)[7]；LSCLEL與LNAL為本文提出的自適應(yīng)局部斥力損失與歸一化面積損失；α與β為平衡各項(xiàng)損失的權(quán)重系數(shù)。

2.1 自適應(yīng)局部斥力損失

在實(shí)際應(yīng)用場(chǎng)景中，由于工程車輛施工區(qū)域較為集中，且工程車輛是在不斷運(yùn)動(dòng)的，導(dǎo)致在施工作業(yè)當(dāng)中，各工程車輛之間容易相互遮擋，當(dāng)多個(gè)工程車輛相互重疊時(shí)，因?yàn)槟繕?biāo)具有相似的外觀特征，檢測(cè)網(wǎng)絡(luò)容易將其混淆。因此，待預(yù)測(cè)的工程車輛的預(yù)測(cè)框可能會(huì)偏移到其他工程車輛，甚至將多個(gè)工程車輛都涵蓋，導(dǎo)致不準(zhǔn)確的定位。如圖2所示，當(dāng)工程車輛A與工程車輛B和C有部分重疊時(shí)，檢測(cè)網(wǎng)絡(luò)很容易將其混淆，導(dǎo)致工程車輛A的預(yù)測(cè)框向工程車輛B和C偏移，致使定位不準(zhǔn)確。

圖2 工程車輛遮擋情況示意Fig.2 Schematic diagram of engineering vehicles occlusion

針對(duì)以上問(wèn)題，本文提出自適應(yīng)局部斥力損失。在自適應(yīng)局部斥力損失中，要求每個(gè)工程車輛預(yù)測(cè)框遠(yuǎn)離其他符合篩選條件的工程車輛的真實(shí)矩形框，同時(shí)自適應(yīng)地調(diào)節(jié)不同情況下的權(quán)重系數(shù)，將遮擋嚴(yán)重的情況賦予更大的損失權(quán)重，其表達(dá)式如式(2)所示：

(2)

式中：G為工程車輛真實(shí)框；P+表示全部的工程車輛正預(yù)測(cè)框，其定義為與至少1個(gè)工程車輛的真實(shí)框的IoU(Intersect Over Union)大于0.5的工程車輛預(yù)測(cè)框的集合，IoU的計(jì)算如式(3)所示：

(3)

(4)

IoG(P,G)的含義為工程車輛預(yù)測(cè)框與工程車輛真實(shí)框的交疊程度在工程車輛真實(shí)框上的占比[13]，其表達(dá)式如式(5)所示：

(5)

綜上，自適應(yīng)局部斥力損失由2個(gè)因素組成：遠(yuǎn)離其他滿足條件的工程車輛真實(shí)框項(xiàng)與自適應(yīng)調(diào)節(jié)項(xiàng)。當(dāng)工程車輛預(yù)測(cè)框被多個(gè)其他工程車輛真實(shí)框遮擋時(shí)，遠(yuǎn)離遮擋程度在閾值條件以上的工程車輛真實(shí)框，同時(shí)，自適應(yīng)調(diào)節(jié)項(xiàng)將遮擋更嚴(yán)重的情況給予更大的懲罰，因此，自適應(yīng)局部斥力損失可以自適應(yīng)地防止預(yù)測(cè)框移動(dòng)到周圍相鄰的遮擋目標(biāo)上，使得檢測(cè)模型對(duì)遮擋目標(biāo)更加魯棒。

為更好地闡述自適應(yīng)局部斥力損失的機(jī)理，圖3為在圖2基礎(chǔ)上，對(duì)自適應(yīng)局部斥力損失的形象化展示。在圖3中，雙向箭頭表示當(dāng)前訓(xùn)練圖像下工程車輛A預(yù)測(cè)框與其他車輛真實(shí)框之間的排斥方向，箭頭的長(zhǎng)度表示其排斥的力度；單向箭頭表示網(wǎng)絡(luò)訓(xùn)練過(guò)程中對(duì)工程車輛A的預(yù)測(cè)框的不同位置的逐漸優(yōu)化的方向，箭頭的長(zhǎng)度表示其網(wǎng)絡(luò)對(duì)預(yù)測(cè)框不同位置優(yōu)化的“力度”。由圖3可知，本文提出的自適應(yīng)局部斥力損失可在網(wǎng)絡(luò)訓(xùn)練過(guò)程中使被檢測(cè)的工程車輛在被遮擋的情況下，使預(yù)測(cè)框與其他的工程車輛真實(shí)框相排斥，使其“收縮”到靠近真實(shí)框的位置，同時(shí)在遮擋越嚴(yán)重的情況下，網(wǎng)絡(luò)會(huì)給予更大的懲罰。

圖3 自適應(yīng)局部斥力損失的機(jī)理示意Fig.3 Schematic diagram of mechanism of self-adaptive local repulsion loss

2.2 歸一化面積損失

由于工程車輛存在形態(tài)的變化(挖掘手在運(yùn)動(dòng))，所以在圖像中會(huì)存在局部特征與全局特征相似的問(wèn)題，這將導(dǎo)致檢測(cè)網(wǎng)絡(luò)易將工程車輛的一部分當(dāng)做目標(biāo)而導(dǎo)致錯(cuò)誤的檢測(cè)。如圖4所示，由于工程車輛D的挖掘手伸展過(guò)開(kāi)，導(dǎo)致挖掘手那部分的特征不明顯，工程車輛D的局部特征與全局特征過(guò)于相似，致使檢測(cè)網(wǎng)絡(luò)的預(yù)測(cè)框從整體偏移到了局部區(qū)域，使定位不準(zhǔn)確。

圖4 工程車輛局部性誤檢示意Fig.4 Schematic diagram of local error detection of engineering vehicles

由圖4可以看出，在工程車輛局部特征與全局特征相似的問(wèn)題中，局部特征區(qū)域的面積往往遠(yuǎn)遠(yuǎn)小于全局面積，所以在本文提出的歸一化面積損失中，要求每個(gè)工程車輛預(yù)測(cè)框與其IoU最大的工程車輛真實(shí)框的面積差盡量小，同時(shí)為了避免面積差的大幅度跳變導(dǎo)致網(wǎng)絡(luò)梯度爆炸，將面積差使用工程車輛真實(shí)框面積進(jìn)行歸一化處理，歸一化面積損失表達(dá)式如式(6)所示：

(6)

式中：G代表工程車輛真實(shí)框；P代表工程車輛預(yù)測(cè)框。

歸一化面積損失將學(xué)習(xí)集中在面積具有相對(duì)較大預(yù)測(cè)誤差的工程車輛上，可對(duì)局部特征與全局特征相似的問(wèn)題進(jìn)行顯著的改進(jìn)。

3 基準(zhǔn)檢測(cè)網(wǎng)絡(luò)介紹與設(shè)定初始錨框優(yōu)化檢測(cè)的精準(zhǔn)度

本文選用經(jīng)典檢測(cè)網(wǎng)絡(luò)Faster R-CNN作為基準(zhǔn)檢測(cè)網(wǎng)絡(luò)；同時(shí)為了在不增加任何成本的情況使網(wǎng)絡(luò)的檢測(cè)框更精細(xì)，使用聚類算法設(shè)定初始錨框優(yōu)化檢測(cè)的精準(zhǔn)度。

3.1 基準(zhǔn)檢測(cè)網(wǎng)絡(luò)Faster R-CNN

Faster R-CNN[7]作為最經(jīng)典的二階段目標(biāo)檢測(cè)框架，相比于一階段方法(如YOLO[14]和SSD[15]等)與無(wú)錨框方法(如CornerNet[16]和FCOS[17]等)，其模塊拆解細(xì)致，對(duì)輸入進(jìn)行特征提取再進(jìn)行區(qū)域建議，再微調(diào)區(qū)域建議網(wǎng)絡(luò)輸出的建議框，可解釋性強(qiáng)，針對(duì)工業(yè)生產(chǎn)可修改空間大，所以本文選擇其作為針對(duì)工程車輛的基準(zhǔn)檢測(cè)網(wǎng)絡(luò)。同時(shí)，對(duì)于一階段方法與無(wú)錨框方法，提出的自適應(yīng)局部斥力與歸一化面積損失函數(shù)等方法也可以作為即插即用的模塊擴(kuò)展到以上方法中。

Faster R-CNN[7]主要由4個(gè)部分組成：特征提取網(wǎng)絡(luò)(Backbone)、區(qū)域建議網(wǎng)絡(luò)(Region Proposal Networks，RPN)、RoI Poling層、分類回歸層。

1)特征提取網(wǎng)絡(luò)：利用一系列卷積、池化、非線性模塊的組合，從淺至深提取輸入圖像的特征并形成特征圖(Feature Map)，為方便闡述，本文選用經(jīng)典的Resnet50網(wǎng)絡(luò)[18]。

2)區(qū)域建議網(wǎng)絡(luò)：通過(guò)預(yù)設(shè)尺寸與比例的多種錨框進(jìn)行特征圖滑動(dòng)并生成多個(gè)候選框，采用非極大值抑制[14](Non-Maximum Suppression,NMS)并篩選出N個(gè)目標(biāo)候選區(qū)域。

3)RoI Poling層：將每個(gè)目標(biāo)候選區(qū)域均勻分成n×n個(gè)塊并進(jìn)行最大池化，得到固定尺度的特征圖候選區(qū)域。

4) 分類回歸層：將RoI Pooling層得到的特征向量再輸入全連接層，以判斷特征圖候選區(qū)域的類別以及準(zhǔn)確位置。

其整體網(wǎng)絡(luò)結(jié)構(gòu)如圖5所示。

圖5 Faster R-CNN網(wǎng)絡(luò)結(jié)構(gòu)Fig.5 Faster R-CNN network structure

3.2 使用聚類算法設(shè)定初始錨框

由于施工場(chǎng)地較大，且工程車輛是在不斷運(yùn)動(dòng)的，導(dǎo)致遠(yuǎn)近工程車輛目標(biāo)尺度相差較大。在基準(zhǔn)檢測(cè)網(wǎng)絡(luò)中使用固定錨框尺度{1282,2562,5122}(px)對(duì)模型進(jìn)行訓(xùn)練與預(yù)測(cè)[7]，會(huì)導(dǎo)致在工程車輛的檢測(cè)中存在部分框回歸偏差的情況。

為了解決遠(yuǎn)近工程車輛目標(biāo)尺度相差較大的問(wèn)題，本文使用K-Means聚類算法[19]優(yōu)化初始錨框參數(shù)。K-Means算法是一種聚類算法，其核心思想是對(duì)于給定的樣本集，按照各樣本間的距離大小，將距離相近的樣本劃分為一類，使類間距離盡量小，類間距離盡量大，共劃分出k類。假設(shè)樣本的類劃分為(C1,C2,…,Ck)，本文采用歐式距離來(lái)度量樣本間的距離，算法目標(biāo)為最小化平方誤差E，E的計(jì)算方法如式(7)所示：

(7)

式中：x為樣本；μi為類Ci的均值向量，也稱為質(zhì)心，其表達(dá)式如式(8)所示：

(8)

本文讀取工程車輛數(shù)據(jù)集中的所有車輛標(biāo)注框的長(zhǎng)與寬，并計(jì)算長(zhǎng)寬比，計(jì)算標(biāo)注框的面積，進(jìn)行K-Means聚類。聚類為5類時(shí)，各聚類簇尺度區(qū)分明顯，效果較好。長(zhǎng)寬尺寸聚類結(jié)果、長(zhǎng)寬比聚類結(jié)果、標(biāo)注框面積聚類結(jié)果如圖6所示。

圖6 多種聚類結(jié)果Fig.6 Multiple clustering results

綜合圖6，可知標(biāo)注框長(zhǎng)寬比集中在0.75～2；標(biāo)注框的面積集中在50 000～700 000 px；再根據(jù)聚類結(jié)果，所以本文調(diào)整初始錨框尺度為{2002,3202,4402,6602,8002}(px)，長(zhǎng)寬比為{0.5,0.67,1,1.5,2}。

4 實(shí)驗(yàn)結(jié)果及討論

4.1 實(shí)驗(yàn)數(shù)據(jù)庫(kù)與實(shí)驗(yàn)環(huán)境

由于目前無(wú)完整的關(guān)于工程車輛在復(fù)雜場(chǎng)景下施工的圖像數(shù)據(jù)集，需要自行收集數(shù)據(jù)集。本文的工程車輛數(shù)據(jù)集是使用大疆無(wú)人機(jī)在國(guó)家電網(wǎng)四川某施工工地采集工程車輛視頻制作的數(shù)據(jù)集，采用人工標(biāo)定工程車區(qū)域作為標(biāo)準(zhǔn)，篩選出包含完整工程車輛的視頻幀，建立總共2 000張帶有標(biāo)準(zhǔn)標(biāo)簽的施工車輛大圖數(shù)據(jù)庫(kù)，圖片分辨率為(1 920×1 080)像素，其中訓(xùn)練集為1 500張，測(cè)試集為500張。

本文實(shí)驗(yàn)硬件配置為：GPU：Nvidia TITAN Xp，12G顯存；CPU：Intel E5-2630。軟件平臺(tái)為Ubuntu 16.04。

4.2 實(shí)驗(yàn)評(píng)價(jià)指標(biāo)

為了驗(yàn)證本文算法的有效性，使用準(zhǔn)確率、mAP值[20]與本文提出的困難場(chǎng)景下的漏檢率(Undetected Rate，UR)、部分檢出率(Partial-detetion Rate，PR)來(lái)作為定位的準(zhǔn)確評(píng)價(jià)指標(biāo)。

4.2.1mAP與準(zhǔn)確率

mAP值為每個(gè)類別AP值的均值，在本文中有3類工程車輛，則有式(9)：

(9)

其中每類工程車輛AP值的計(jì)算由其精確率(Precision)與召回率(Recall)確定，精確率是指被識(shí)別為某類工程車輛的樣本占真正某類工程車輛樣本的比例；召回率是指被識(shí)別出的某類真正工程車輛的樣本占所有某類工程車輛樣本的比重，其計(jì)算公式如式(10)所示：

(10)

式中：TP表示被正確識(shí)別出的某類工程車輛樣本數(shù)；FP表示被誤識(shí)別為其他類工程車輛的某類工程車輛樣本數(shù)；FN表示未被識(shí)別出的某類工程車輛樣本數(shù)。對(duì)于同一個(gè)模型，IoU與置信度是影響精確率與召回率的值的2個(gè)關(guān)鍵因素[18]。對(duì)于不同的IoU閾值與置信度閾值，召回率和精確度會(huì)有所變化。本文取IoU閾值為0.7，對(duì)于不同置信度閾值，將精確率與召回率分別作為橫、縱坐標(biāo)，可以得到P-R曲線，曲線與橫、縱軸圍成的面積即為AP值。另外，本文取IoU閾值為0.7下的準(zhǔn)確率作為本文的準(zhǔn)確率評(píng)估指標(biāo)。

4.2.2 漏檢率與部分檢出率

為了更直觀地驗(yàn)證本文算法在解決遮擋問(wèn)題與特征相似性時(shí)的有效性，提出困難場(chǎng)景下的漏檢率與部分檢出率2個(gè)評(píng)價(jià)指標(biāo)來(lái)評(píng)價(jià)算法的性能。

1)對(duì)于漏檢率，本文定義漏檢為：對(duì)于1個(gè)工程車輛真實(shí)框G，使用這個(gè)真實(shí)工程車輛框圖像的所有預(yù)測(cè)框與其計(jì)算IoU，若所有預(yù)測(cè)框與其IoU都小于0.1，則定義這個(gè)真實(shí)車輛框存在漏檢情況。其表達(dá)式如式(11)所示：

G∈GUs.t.?P,IoU(P,G)<0.1

(11)

式中：GU表示漏檢框的集合；P表示預(yù)測(cè)框。則漏檢率為所有漏檢的工程車輛真實(shí)框數(shù)目占所有工程車輛真實(shí)框數(shù)目的比例，如式(12)所示：

(12)

2)對(duì)于部分檢出率，本文定義部分檢出為：對(duì)于1個(gè)工程車輛真實(shí)框G，使用這個(gè)真實(shí)工程車輛框圖像的所有預(yù)測(cè)框與其計(jì)算IoU，找出與其IoU最大的那個(gè)預(yù)測(cè)框P，若滿足0.1

(13)

式中：各物理量的含義與式(11)一致。則部分檢出率為所有部分檢出的工程車輛真實(shí)框數(shù)目占所有工程車輛真實(shí)框數(shù)目的比例，如式(14)所示：

(14)

4.3 工程車輛目標(biāo)檢測(cè)實(shí)驗(yàn)與分析

為了進(jìn)一步驗(yàn)證本文提出算法的有效性，在本文構(gòu)建的工程車輛數(shù)據(jù)集基礎(chǔ)上，分別用基準(zhǔn)檢測(cè)網(wǎng)絡(luò)Faster R-CNN與融合本文提出算法的基準(zhǔn)檢測(cè)網(wǎng)絡(luò)進(jìn)行訓(xùn)練與測(cè)試，同時(shí)進(jìn)行各個(gè)子算法排列的消融實(shí)驗(yàn)，使用測(cè)試集進(jìn)行檢測(cè)性能測(cè)試對(duì)比。實(shí)驗(yàn)的評(píng)價(jià)結(jié)果如表1所示。

由表1可知，融合了本文提出的歸一化面積損失與自適應(yīng)局部斥力損失的基礎(chǔ)檢測(cè)模型在數(shù)據(jù)庫(kù)測(cè)試集進(jìn)行檢測(cè)的總體情況在漏檢率、部分檢出率、各類別準(zhǔn)確率、mAP指標(biāo)上均好于其他算法。盡管YOLOv3[14]與SSD[13]算法速度較快，但是其精度較低，而本文算法22 FPS的速度已經(jīng)能滿足施工場(chǎng)景下的監(jiān)管需要。同時(shí)，由于本文的數(shù)據(jù)采集來(lái)自不同的時(shí)間段，可以覆蓋不同的光照?qǐng)鼍?，所以總體的實(shí)驗(yàn)結(jié)果可以代表不同光照下的準(zhǔn)確率。按總體實(shí)驗(yàn)結(jié)果來(lái)看，在不同光照下，本文的算法在mAP指標(biāo)上達(dá)到了91.3%，裝載機(jī)的準(zhǔn)確率達(dá)到88.4%，其余類別工程車輛準(zhǔn)確率均超過(guò)90%。漏檢率指標(biāo)尤其能表示工程車輛之間遮擋情況下的模型性能，而在漏檢率指標(biāo)上，融合了本文提出的歸一化面積損失與自適應(yīng)局部斥力損失的基礎(chǔ)檢測(cè)模型的漏檢率指標(biāo)超過(guò)其他算法近6個(gè)百分點(diǎn)，證明了本文提出的自適應(yīng)局部斥力損失的有效性。部分檢出率指標(biāo)尤其能表示在工程車輛自身特征相似性較高情況下的模型性能，而在部分檢出率指標(biāo)上，融合了本文提出的歸一化面積損失與自適應(yīng)局部斥力損失的基礎(chǔ)檢測(cè)模型的部分檢出率指標(biāo)超過(guò)其他算法近4個(gè)百分點(diǎn)，證明了本文提出的歸一化面積損失的有效性。

表1 各算法工程車輛目標(biāo)檢測(cè)對(duì)比結(jié)果Table 1 Comparison results of target detection on engineering vehicles by each algorithm

無(wú)遮擋情況下的不同檢測(cè)模型目標(biāo)檢測(cè)示意圖如圖7所示，圖7(a)和圖7(b)分別為原始基礎(chǔ)檢測(cè)模型與融合本文提出算法的基礎(chǔ)檢測(cè)模型的目標(biāo)檢測(cè)定位效果。圖7(a)中，被原始基礎(chǔ)檢測(cè)模型將工程車輛的一部分當(dāng)作目標(biāo)而導(dǎo)致錯(cuò)誤的檢測(cè)，在圖7(b)中被正確的識(shí)別，說(shuō)明本文所提出的歸一化面積損失，對(duì)由于工程車輛局部特征與全局特征相似導(dǎo)致算法易將工程車輛的一部分當(dāng)作目標(biāo)導(dǎo)致錯(cuò)誤的檢測(cè)這個(gè)問(wèn)題有了很大的改善。同時(shí)，也可看出部分漏檢在融合本文提出的算法基礎(chǔ)檢測(cè)模型中被正確的檢測(cè)到，有部分框回歸偏差也被修正的更佳，證明了本文改進(jìn)的聚類、數(shù)據(jù)增強(qiáng)的有效性。

圖7 2種模型的工程車輛目標(biāo)檢測(cè)結(jié)果對(duì)比Fig.7 Comparison on target detection results of engineering vehicles by two models

遮擋嚴(yán)重情況下的目標(biāo)檢測(cè)示意圖如圖8所示。圖8(a)和圖8(b)分別為遮擋嚴(yán)重情況下的原始基礎(chǔ)檢測(cè)模型與融合本文提出算法的基礎(chǔ)檢測(cè)模型的定位效果。在圖8(a)中，當(dāng)多個(gè)工程車輛相互遮擋時(shí)，待預(yù)測(cè)的工程車輛的預(yù)測(cè)框偏移到其他工程車輛，甚至將多個(gè)工程車輛都涵蓋，而這個(gè)問(wèn)題在圖8(b)中得到了很好的改善，在待預(yù)測(cè)的工程車輛被遮擋情況下，預(yù)測(cè)框不會(huì)再偏移到其他工程車輛，具有良好的表現(xiàn)，說(shuō)明本文所提出的自適應(yīng)局部斥力損失對(duì)工程車輛的遮擋問(wèn)題具有良好的性能。

圖8 遮擋嚴(yán)重下的工程車輛目標(biāo)檢測(cè)結(jié)果示意Fig.8 Schematic diagram of target detection results of engineering vehicles under severe occlusion

5 結(jié)論

1)針對(duì)施工場(chǎng)景下工程車輛易互相遮擋，工程車輛局部特征與全局特征相似以及場(chǎng)景環(huán)境復(fù)雜的問(wèn)題，提出一種基于自適應(yīng)局部斥力與歸一化面積損失的工程車輛的目標(biāo)檢測(cè)算法。

2)以Faster R-CNN為例作為基礎(chǔ)檢測(cè)模型，針對(duì)遮擋問(wèn)題與特征相似性設(shè)計(jì)了自適應(yīng)局部斥力損失與歸一化面積損失，并結(jié)合聚類算法設(shè)定初始錨框減少預(yù)測(cè)框的偏移。

3)實(shí)驗(yàn)結(jié)果表明，檢測(cè)結(jié)果mAP達(dá)到91.3%，采用該算法可以很好地解決工程車輛遮擋、工程車輛特征相似以及場(chǎng)景環(huán)境復(fù)雜3方面問(wèn)題。