基于改進(jìn)YOLOv7的紅外安防目標(biāo)檢測
激光雜志
頁數(shù): 7 2024-05-25
摘要: 針對安防場景中紅外圖像對比度低、目標(biāo)輪廓不明顯導(dǎo)致目標(biāo)檢測效果差的問題,提出一種基于改進(jìn)YOLOv7的紅外安防目標(biāo)檢測算法。采用遞歸門控卷積改進(jìn)主干網(wǎng)絡(luò),增強(qiáng)對輸入圖像高階信息交互能力;使用SimAM注意力機(jī)制構(gòu)建ELAN-S模塊,降低信息丟失率的同時減少網(wǎng)絡(luò)參數(shù);使用K-means++聚類算法優(yōu)化錨盒尺寸,提高檢測精度。對InfiRay公開數(shù)據(jù)集進(jìn)行數(shù)據(jù)增強(qiáng)和模型驗證實驗,結(jié)...