基于Snake与注意力机制的船舶实例分割方法

Ship Contour: a novel instance segmentation approach based on Snake and Attention Mechanism

  • 摘要: 摘 要:【目的】船舶的实例分割可服务于船舶监测、识别和跟踪等任务,支撑智能航行。然而,受船舶形状尺度多变、环境因素干扰,已有实例分割方法在船舶轮廓提取上表现不佳。为此,本文提出基于曲线递归的Ship Contour方法。【方法】通过改进CenterNet方法提取分层特征,融合Deep Layer Aggregation-60骨干网络,兼顾精度和速度;改进Block结构、引入ECA通道注意力机制增强特征提取的能力,使用Mish激活函数代替ReLU适应深层学习;引入平移不变的轮廓变形方法、Dynamic Matching Loss损失函数加速最终轮廓的提取。【结果】在2300张样本的2023Ship-seg专用数据集上,所提出方法的准确率AP0.5:0.95达到了64.0%,召回率AR0.5:0.95到达了67.9%,优于所有主流实例分割算法。【结论】方法将有效提升监控与智能航行场景下的视觉处理效果。

     

    Abstract: Abstract: The instance segmentation of ships can serve tasks such as ship monitoring, identification, and tracking, supporting intelligent navigation. However, due to the variability in ship shapes and scales, as well as interference from environmental factors, existing instance segmentation methods perform poorly in ship contour extraction. Therefore, this paper proposes a Ship Contour method based on curve recursion. By improving the CenterNet method to extract hierarchical features and integrating the Deep Layer Aggregation-60 backbone network, the method balances accuracy and speed; the Block structure is improved, and the ECA channel attention mechanism is introduced to enhance the ability of feature extraction, using the Mish activation function instead of ReLU to adapt to deep learning; a translation-invariant contour deformation method and a Dynamic matching loss function are introduced to accelerate the extraction of the final contour. On the 2023Ship-seg dedicated dataset with 2300 samples, the accuracy AP0.5:0.95 of the proposed method reached 64.0%, and the recall rate AR0.5:0.95 reached 67.9%, outperforming all mainstream instance segmentation algorithms. The method will effectively enhance the visual processing effects in monitoring and intelligent navigation scenarios.

     

/

返回文章
返回