基于ResNet50和改進注意力機制的船舶識別模型
計算機應(yīng)用
頁數(shù): 7 2023-09-05
摘要: 海洋船舶的自動識別對緩解海上交通壓力起著重要作用。針對當(dāng)前船舶自動識別率較低的問題,提出一種基于ResNet50(Residual Network50)和改進注意力機制的船舶識別模型。首先,自制船舶數(shù)據(jù)集并劃分為訓(xùn)練集、驗證集和測試集,采用模糊、增加噪聲等方法得到增強數(shù)據(jù)集;其次,設(shè)計改進注意力模塊——高效空間金字塔注意力模塊(ESPAM)和船舶種類識別模型ResNet50_E... (共7頁)