一種基于YOLOv5s的紅外圖像目標(biāo)檢測(cè)改進(jìn)算法
激光與紅外
頁(yè)數(shù): 9 2023-07-20
摘要: 受熱紅外成像方式限制,交通場(chǎng)景下紅外圖像存在對(duì)比度低、目標(biāo)尺度和姿態(tài)的多樣性以及目標(biāo)之間的相互遮擋問(wèn)題,從而造成檢測(cè)精度下降,部分目標(biāo)出現(xiàn)漏檢、誤檢的情況。本文在YOLOv5s的基礎(chǔ)上提出一種改進(jìn)算法:在數(shù)據(jù)處理方面,使用AHE算法對(duì)訓(xùn)練集圖像進(jìn)行部分?jǐn)?shù)據(jù)增強(qiáng);在模型改進(jìn)方面,通過(guò)引入跨域遷移學(xué)習(xí)策略、插入通道注意力機(jī)制SENet、改進(jìn)損失函數(shù)GIoU為α-CIoU對(duì)YOLOv...