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