基于改進(jìn)YOLOv5的無(wú)人船視角目標(biāo)檢測(cè)方法

摘要： 針對(duì)復(fù)雜海況下無(wú)人船小尺度水面目標(biāo)檢測(cè)效果不佳及無(wú)人船數(shù)據(jù)集出現(xiàn)的樣本不均衡的情況，本文提出一種改進(jìn)YOLOv5的無(wú)人船水面目標(biāo)檢測(cè)方法。首先，將YOLOv5中主干網(wǎng)絡(luò)部分替換為GhostNet輕量化網(wǎng)絡(luò)。其次，在特征提取網(wǎng)絡(luò)中加入全局空間自適應(yīng)模塊，以限制檢測(cè)圖空間維度上的影響。最后，損失函數(shù)部分基于多分類交叉熵?fù)p失函數(shù)加入可訓(xùn)練的噪聲參數(shù)，降低樣本不平衡帶來(lái)的影響。通過無(wú)人...