马枫, 石子慧, 孙杰, 等. 自注意力机制驱动的轻量化高鲁棒船舶目标检测方法[J]. 中国舰船研究, 2024, 19(X): 1–12. DOI: 10.19693/j.issn.1673-3185.03389
引用本文: 马枫, 石子慧, 孙杰, 等. 自注意力机制驱动的轻量化高鲁棒船舶目标检测方法[J]. 中国舰船研究, 2024, 19(X): 1–12. DOI: 10.19693/j.issn.1673-3185.03389
MA F, SHI Z H, SUN J, et al. Lightweight and robust ship detection method driven by self-attention mechanism[J]. Chinese Journal of Ship Research, 2024, 19(X): 1–12 (in Chinese). DOI: 10.19693/j.issn.1673-3185.03389
Citation: MA F, SHI Z H, SUN J, et al. Lightweight and robust ship detection method driven by self-attention mechanism[J]. Chinese Journal of Ship Research, 2024, 19(X): 1–12 (in Chinese). DOI: 10.19693/j.issn.1673-3185.03389

自注意力机制驱动的轻量化高鲁棒船舶目标检测方法

Lightweight and robust ship detection method driven by self-attention mechanism

  • 摘要:
    目的 海岸监控与驾驶瞭望过程中,需要在远距离、多场景下对各种目标进行识别与跟踪。其中,船舶目标往往成像小、特征不明显,容易与其他目标混淆。为此,提出一种船舶检测方法ShipDet,通过设计专用骨干网络、改进特征提取过程、约束微观检测头,旨在改善上述问题。
    方法 首先,通过融合自注意力模块Swin Transformer(STR)和经典CSPDarknet53网络,构造对微小物标高度敏感的特征融合提取网络,以增强小目标特征与环境的相关关系,建立船与航道、船与船、船与岸线的关联,显著抑制不相关信息。然后,考虑到数据集中船舶目标分布不均匀并且尺度变化较小的特点,保留2个检测层,减少模型参数并进一步提升模型性能。最后,使用SIoU损失函数(SCYLLA-IoU)来约束检测头,降低损失函数的回归自由度,提高检测的精度和抗干扰能力。
    结果 在2023ships数据集上的验证结果表明,提出的方法在船舶目标检测任务上表现较好,mAP@0.5达到了92.9%,平均精度为92.1%,消耗参数量仅为35 366 310,整体检测性能优于其他算法。
    结论 ShipDet方法将为海事监控、智能航行提供高效的支撑。

     

    Abstract:
    Objective It is vital to detect and track ships during coastal monitoring and ship navigation over long distances in complex circumstances. However, due to their small size and unclear features, they can be readily confused with shorelines, noise, and rocks, making them sometimes difficult to spot immediately. To address this issue, a novel ship detection method called ShipDet is proposed which significantly improves performance through the design of a dedicated backbone network, improved feature extraction process, and constrained microscopic detection heads.
    Method First, this method constructs a feature fusion and extraction network that is highly sensitive to small objects by integrating the Swin Transformer module (STR) with the classic CSPDarknet53 network. This enhances the correlation between small target features and the environment, establishing associations between ships and waterways, ships and other ships, and ships and coastlines, while suppressing irrelevant information. Subsequently, considering the uneven distribution and minor scale variations of ship targets in the dataset, two detection layers are retained to reduce model parameters and further enhance model performance. Moreover, the method employs the SCYLLA-IoU (SIoU) loss function to constrain the detection heads, thereby reducing regression freedom and improving detection accuracy and robustness.
    Results To validate the proposed method, a dataset called 2023ships is established which consists of up to 9,000 samples covering various scenarios such as inland rivers, coastal areas, daytime, nighttime, and foggy weather. During testing, the proposed method demonstrates superior overall ship detection performance compared to other algorithms, with a mAP of 92.9%, a precision rate of 92.1%, and a parameter size of 35,366,310.
    Conclusion The proposed method can greatly benefit the fields of maritime monitoring and intelligent navigation.

     

/

返回文章
返回