基于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损失函数加快最终轮廓的提取。
    结果 在2 300张样本的2023Ship-seg专用数据集上,所提出方法的准确率AP0.5∶0.95达到64.0%,召回率AR0.5∶0.95达到67.9%,优于主流实例分割算法。
    结论 所提方法能有效提升监控与智能航行场景下的视觉处理效果。

     

    Abstract:
    Objective Instance segmentation of ships plays a crucial role in tasks such as monitoring, identification, and tracking, thereby supporting intelligent navigation. However, the wide variability in ship shapes and scales, coupled with environmental interference, leads to poor performance of existing methods perform in Ship Contour extraction. To address this issue, this paper proposes a novel Ship Contour method based on curve recursion.
    Method  By enhancing CenterNet with hierarchical feature extraction and integrating the Deep Layer Aggregation-60 backbone network, the proposed method achieves a balance between accuracy and speed. The Block structure is optimized, and an ECA channel attention mechanism is incorporated to strengthen feature extraction, while the Mish activation function replaces ReLU to improve adaptability in deep learning. In addition, a translation-invariant contour deformation method and a dynamic matching loss function are introduced to accelerate the final contour extraction.
    Results  On the dedicated 2023Ship-seg dataset containing 2300 samples, the proposed method achieved an average precision of AP0.5∶0.95 = 64.0% and a recall rate of AR0.5∶0.95 = 67.9%, outperforming all mainstream instance segmentation algorithms.
    Conclusion  The method can significantly improve visual processing performance in ship monitoring and intelligent navigation scenarios.

     

/

返回文章
返回