基于改進YOLOv5的無人船視角目標檢測方法

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