摘要:路面坑洼是主要道路缺陷,會(huì )損壞車(chē)輛,影響駕駛員的安全駕駛,嚴重時(shí)還會(huì )導致交通事故,針對這個(gè)問(wèn)題,本文提出了改進(jìn)YOLOv7的道路坑洼檢測算法。首先,使用Mosaic+Mixup進(jìn)行內置數據增強,擴充小樣本數據集,增強模型泛化能力;其次,引入CA注意力機制,將橫縱位置信息編碼,保證計算量的同時(shí)又能關(guān)注大范圍位置信息;然后,引入BIFPN雙向特征金字塔網(wǎng)絡(luò ),通過(guò)特征融合多尺度語(yǔ)義特征提高檢測效率;最后,將損失函數SIoU替換CIoU,有效解決回歸中的樣本不平衡問(wèn)題。實(shí)驗結果表明,改進(jìn)之后的算法在坑洼數據集的平精度均值和精確率達到了89.42%和90.12%,相比于原本的YOLOv7版本提高了6.18%和1.96%,更準確更快速的應用于道路維修。