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