基于注意力機(jī)制及多尺度融合的紅外船舶檢測(cè)
激光與光電子學(xué)進(jìn)展
頁數(shù): 7 2023-11-20
摘要: 針對(duì)紅外圖像船舶目標(biāo)檢測(cè)方法的準(zhǔn)確性和實(shí)時(shí)性還不能滿足海防場(chǎng)景需求的問題,提出一種基于改進(jìn)YOLOv7算法的輕量級(jí)船舶檢測(cè)算法。該算法首先在Backbone網(wǎng)絡(luò)中引入MobileNetv3主干網(wǎng)絡(luò),實(shí)現(xiàn)模型輕量化處理。然后在Neck網(wǎng)絡(luò)引入注意力機(jī)制抑制噪聲與干擾,以提高網(wǎng)絡(luò)的特征提取能力,并采用雙向加權(quán)特征金字塔,以提升特征融合能力。最后引入Wise IoU優(yōu)化損失函數(shù),提高...