賈睿JIA Rui;趙紅巖ZHAO Hong-yan
(遼寧省交通高等專(zhuān)科學(xué)校,沈陽(yáng) 110122)
快速發(fā)展的道路交通基礎(chǔ)設(shè)施為人們?nèi)粘I钐峁┍憷?,同時(shí)也給道路病害檢測(cè)和養(yǎng)護(hù)帶來(lái)了新的挑戰(zhàn)。隨著我國(guó)交通運(yùn)輸量的持續(xù)增長(zhǎng),路面在交通設(shè)備的壓力下,會(huì)產(chǎn)生各種不同類(lèi)型、不同程度的病害。若道路病害沒(méi)有被及時(shí)發(fā)現(xiàn)并處理,會(huì)影響車(chē)輛行駛安全、減少道路設(shè)備的使用年限、增加交通養(yǎng)護(hù)成本,還有很大可能引發(fā)交通事故[1]。目前,傳統(tǒng)道路巡檢主要采用數(shù)據(jù)采集+專(zhuān)家判斷的形式,其檢測(cè)準(zhǔn)確性依賴于設(shè)備精準(zhǔn)度和人員水平,存在巡檢時(shí)間長(zhǎng)、養(yǎng)護(hù)成本高、人工依賴度高、檢測(cè)精準(zhǔn)率低等問(wèn)題[2]。因此使用先進(jìn)技術(shù)手段,對(duì)道路病害進(jìn)行快速、準(zhǔn)確、智能檢測(cè)具有重大意義,成為道路巡檢養(yǎng)護(hù)管理的發(fā)展趨勢(shì)。
以產(chǎn)業(yè)數(shù)字化為發(fā)展趨勢(shì)的時(shí)代背景下,道路養(yǎng)護(hù)行業(yè)在使用數(shù)字化技術(shù)來(lái)實(shí)現(xiàn)高效準(zhǔn)確的道路病害智能巡檢和養(yǎng)護(hù)。隨著人工智能和計(jì)算機(jī)視覺(jué)的發(fā)展,很多科研人員將深度學(xué)習(xí)將卷積神經(jīng)網(wǎng)絡(luò)引入道路病害檢測(cè)領(lǐng)域,取得了良好成效,提升病害識(shí)別性能,降低檢測(cè)成本。
不同于傳統(tǒng)圖像處理中需要人為設(shè)計(jì)特征的模式,深度卷積神經(jīng)網(wǎng)絡(luò)具有特征自動(dòng)發(fā)現(xiàn)功能。因此,基于深度卷積神經(jīng)網(wǎng)絡(luò)的道路病害檢測(cè)算法能夠自主進(jìn)行病害特征學(xué)習(xí)和識(shí)別,具有檢測(cè)準(zhǔn)確率高、環(huán)境適應(yīng)性強(qiáng)、時(shí)效高等特點(diǎn)。相關(guān)研究成果已應(yīng)用于道路養(yǎng)護(hù)巡檢、路面完整性驗(yàn)證等領(lǐng)域,帶動(dòng)道路巡檢與道路養(yǎng)護(hù)模式的轉(zhuǎn)型,通過(guò)技術(shù)創(chuàng)新推動(dòng)我國(guó)道路安全巡檢技術(shù)的產(chǎn)業(yè)升級(jí)。
目前,主流路面病害檢測(cè)算法有兩種:傳統(tǒng)圖像處理算法和深度卷積神經(jīng)網(wǎng)絡(luò)算法。
傳統(tǒng)圖像處理算法首先由專(zhuān)家根據(jù)經(jīng)驗(yàn)來(lái)定義病害的圖像特征,再利用此圖像特征進(jìn)行病害檢測(cè)、定位和評(píng)估。Wang[3]使用模板匹配和濾波算法來(lái)識(shí)別公路路面裂縫病害。Hu[4]提出了局部二值模式算子對(duì)裂縫病害進(jìn)行識(shí)別。DIXIT[5]使用Sobel算子提取病害的梯度特征。SALMAN[6]利用邊緣檢測(cè)算法來(lái)提取病害特征。然而,由于需要人工設(shè)計(jì)各類(lèi)型病害特征,傳統(tǒng)圖像處理算法容易受光照、陰影和路面障礙等環(huán)境因素影響,檢測(cè)準(zhǔn)確率不高,難以應(yīng)對(duì)復(fù)雜多變的不同道路病害。
不同于傳統(tǒng)圖像處理中需要人為設(shè)計(jì)特征的模式,深度卷積神經(jīng)網(wǎng)絡(luò)具有特征自動(dòng)發(fā)現(xiàn)功能。沙愛(ài)民[7]使用卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行病害分離,此方法采用滑動(dòng)窗口方式進(jìn)行圖處理,將整體圖像切分為不同子塊,然后將圖像子塊輸入神經(jīng)網(wǎng)絡(luò),根據(jù)分類(lèi)結(jié)果以判斷輸入圖像子塊是否是病害。MAEDA[8]將端到端的目標(biāo)檢測(cè)算法應(yīng)用于路面病害檢測(cè)。田楊[9]采用Fast-RCNN算法來(lái)自動(dòng)化識(shí)別坑槽、裂縫等的道路病害信息。孫朝云[10]改進(jìn)SSD目標(biāo)檢測(cè)模型,使用殘差模塊來(lái)進(jìn)行路面裂縫特征提取,并在提取的特征圖上進(jìn)行檢測(cè),識(shí)別道路裂縫。許慧青[11]針對(duì)細(xì)長(zhǎng)路面病害的語(yǔ)義特性和形狀屬性,提出雙階段路面病害檢測(cè)方法Epd RCNN。羅暉[12]針對(duì)路面病害存在種類(lèi)多、形狀多樣、數(shù)據(jù)量不足等問(wèn)題,導(dǎo)致病害檢測(cè)難度大的現(xiàn)狀,提出基于YOLOv4的路面多尺度病害檢測(cè)算法。
3.1 YOLOv5網(wǎng)絡(luò)模型
YOLOv5是一種單階段目標(biāo)檢測(cè)算法,在前序版本基礎(chǔ)上增加了若干改進(jìn)點(diǎn)及優(yōu)化技巧,使YOLOv5算法的運(yùn)行性能和檢測(cè)準(zhǔn)確率得到很大提升。YOLOv5算法模型主要分為四個(gè)模塊:輸入端、骨干網(wǎng)絡(luò)、Neck網(wǎng)絡(luò)與Head輸出端,其中骨干網(wǎng)絡(luò)采用CSPDarknet53網(wǎng)絡(luò)結(jié)構(gòu)。輸入端采用Mosaic數(shù)據(jù)增強(qiáng)技術(shù),對(duì)輸入檢測(cè)圖像進(jìn)行各種變換操作,并提出自適應(yīng)錨框計(jì)算與自適應(yīng)圖片縮放方法,如圖1所示。骨干網(wǎng)絡(luò)使用了CSPDarknet53網(wǎng)絡(luò)結(jié)構(gòu),并融合Focus結(jié)構(gòu),用于提取圖像特征表示。Neck網(wǎng)絡(luò)采用SPP模塊、FPN+PAN模塊,其位于骨干網(wǎng)絡(luò)和頭網(wǎng)絡(luò)的中間,利用它可以進(jìn)一步提升特征的多樣性及魯棒性。Head輸出端完成目標(biāo)檢測(cè)結(jié)果的輸出,利用GIOU_Loss作為損失函數(shù),能夠解決病害圖像和背景圖像不均衡問(wèn)題,從而進(jìn)一步提升算法的檢測(cè)精度。
3.2 Attention-YOLOv5模型
為了更好識(shí)別道路病害,在檢測(cè)網(wǎng)絡(luò)模型中引入通道注意力模塊SE Block[13],通過(guò)學(xué)習(xí)的方式來(lái)自動(dòng)學(xué)習(xí)到各個(gè)特征通道的重要程度,對(duì)各通道進(jìn)行加權(quán)運(yùn)算,使模型更加關(guān)注信息量最大的特征通道,抑制那些不重要的特征通道,從而提升病害檢測(cè)準(zhǔn)確度。通道注意力模塊結(jié)構(gòu)示意圖,如圖2所示。
同時(shí),將增加注意力權(quán)重后的特征傳遞至下一層特征識(shí)別模塊,利用高層注意力信息和低層注意力信息,實(shí)現(xiàn)多尺度圖像特征融合,增加多類(lèi)型多尺度病害檢測(cè)的魯棒性,更好地提取病害特征和病害位置。
本文采使用GRDDC 2020數(shù)據(jù)集,來(lái)源為全球道路病害檢測(cè)挑戰(zhàn)者,數(shù)據(jù)集中包含來(lái)自捷克、日本、印度等不同地區(qū)的真實(shí)道路病害圖片,主要標(biāo)識(shí)了橫向裂紋、縱向裂紋、龜裂紋、坑洼等常見(jiàn)道路病害。由于數(shù)據(jù)來(lái)自于全球不同地區(qū),包括發(fā)達(dá)國(guó)家和發(fā)展中國(guó)家,因此路面硬化的質(zhì)量各不相同,且每種損壞類(lèi)型的損壞程度、面積、長(zhǎng)寬比等差異巨大,代表了各種類(lèi)型路面病害的真實(shí)情況。
為驗(yàn)證本文算法的有效性,比較道路病害檢測(cè)模型的性能和準(zhǔn)確性,在道路病害數(shù)據(jù)集上進(jìn)行了實(shí)驗(yàn),將提出的改進(jìn)模型結(jié)構(gòu)與原始YOLOv5檢測(cè)模型進(jìn)行實(shí)驗(yàn)比較,在同一運(yùn)行環(huán)境下進(jìn)行對(duì)比分析,如表1所示。采用的實(shí)驗(yàn)配置如下:CPU為Intel至強(qiáng)W-2150B,主頻為3.0GHz,內(nèi)存為64GB;GPU為Nvidia GTX 2080Ti,顯存為11GB;PyTorch 1.7.1,CUDA 11.4。
表1 原始模型與改進(jìn)模型效果數(shù)據(jù)對(duì)比
更加直觀地驗(yàn)證病害檢測(cè)模型性能,利用帶有注意力機(jī)制的YOLOv5檢測(cè)模型在數(shù)據(jù)測(cè)試集上進(jìn)行可視化檢測(cè),結(jié)果如圖3所示。
可以看出改進(jìn)后的Attention-YOLOv5檢測(cè)模型能夠檢測(cè)出大部分病害,識(shí)別出病害類(lèi)型,標(biāo)出病害位置信息,且在不同光照環(huán)境下也能給出良好的檢測(cè)結(jié)果。但檢測(cè)結(jié)果出現(xiàn)一定漏檢和誤檢現(xiàn)象,尤其是對(duì)一些小道路病害,檢測(cè)結(jié)果還有待進(jìn)一步改進(jìn),這是由于YOLO檢測(cè)模型屬于單階段檢測(cè)算法,其對(duì)于小目標(biāo)物體檢測(cè)存在不足,后續(xù)考慮增加多層級(jí)特征融合來(lái)進(jìn)一步提升檢測(cè)精準(zhǔn)度。
本文針對(duì)道路病害智能檢測(cè)應(yīng)用,提出一種Attention-YOLOv5檢測(cè)模型,該模型引入通道注意力模塊,將其融入YOLOv5目標(biāo)檢測(cè)模型中,利用通道注意力模塊為多層級(jí)病害特征賦予權(quán)重,弱化背景和噪聲影響,提升病害檢測(cè)性能,增加不同類(lèi)型病害檢測(cè)的魯棒性。實(shí)驗(yàn)數(shù)據(jù)表明本文提出的病害檢測(cè)算法模型具有很好的泛化性能,可廣泛用于道路養(yǎng)護(hù)和巡檢等實(shí)際場(chǎng)景中,推進(jìn)公路交通智慧養(yǎng)護(hù)的發(fā)展。在后續(xù)的工作中可近一步提升算法特征提取能力,優(yōu)化模型檢測(cè)速度。