欢迎访问《兵工学报》官方网站,今天是

兵工学报

• •    下一篇

基于改进DeepLabV3+的海上船舶目标分割方法

陶维昊1,罗亚松1*,林法君2**,曲建静3,刘一平1   

  1. 1. 海军工程大学; 2. 海军参谋部军事训练中心; 3. 久之洋红外系统股份有限公司
  • 收稿日期:2024-09-29 修回日期:2025-03-30
  • 基金资助:
    国家自然科学基金项目(42074074)

Marine Ship Target Segmentation Method Based on Improved DeepLabV3+

TAO Weihao1, LUO Yasong1*, LIN Fajun2**, QU Jianjing3, LIU Yiping1   

  1. 1. Naval University of Engineering; 2. Military Training Center of Naval Staff Department; 3. Jiuzhiyang Infrared System Co., Ltd.
  • Received:2024-09-29 Revised:2025-03-30

摘要: 船舶目标分割对于提高海洋监测系统的自动化程度具有重要意义。现有的分割方法往往存在误分割、易受噪声干扰、处理效率低等问题,难以灵活高效地适应实际场景的需求。针对海上船舶目标分割任务,提出一种基于改进DeepLabV3+网络的分割算法MET-DeepLabV3+。在图像预处理阶段采了线性变换、双边滤波和多尺度Retinex级联式算法,以减弱噪声、天气等不利因素的影响;模型设计中采用轻量化的MobileNetV2作为主干网络,以降低模型复杂度;引入ECA-Net注意力机制,以增强模型对多尺度特征的捕捉能力。采用迁移学习方法,将在SeaShips数据集上预训练得到的特征权重应用于模型训练中,进一步优化分割效果。实验结果表明,改进算法的平均交互比为91.62%,平均像素准确率为92.87%,相比于基础DeepLabV3+模型分别提高2.13%和1.11%,且高于HRNet、PSPNet和UNet等分割模型,较好地满足了船舶目标分割任务的实际需求,具有较高的应用价值。

关键词: 船舶目标分割, DeepLabV3+, 轻量化网络, 注意力机制, 迁移学习

Abstract: Ship target segmentation plays a crucial role in enhancing the automation level of marine monitoring systems. However, existing segmentation methods often face issues such as mis-segmentation, susceptibility to noise interference, and low processing efficiency, making them difficult to adapt flexibly and efficiently to real-world scenarios. To address these challenges in maritime ship target segmentation tasks, a segmentation algorithm is proposed based on an improved DeepLabV3+ network, termed MET-DeepLabV3+. During the image preprocessing stage, bilateral filtering, multi-scale Retinex algorithm, and linear transformation are employed to mitigate the effects of noise, weather, and other adverse factors. In terms of model design, the lightweight MobileNetV2 is used as the backbone network to reduce model complexity, and the ECA-Net attention mechanism is introduced to enhance the model's ability to capture multi-scale features. Additionally, a transfer learning approach is adopted, where feature weights pre-trained on the SeaShips dataset are applied during model training to further optimize segmentation performance. Experimental results demonstrate that the improved algorithm achieves an average interaction-over-union (IoU) of 91.62% and an average pixel accuracy (PA) of 92.87%, representing improvements of 2.13% and 1.11% over the baseline DeepLabV3+ model, respectively. Moreover, it outperforms segmentation models such as HRNet, PSPNet, and UNet, which effectively meets practical demands of ship target segmentation tasks and offering significant application value.

Key words: ship target segmentation, DeepLabV3+, lightweight network, attention mechanism, transfer

中图分类号: