改进YOLOv11n的轻量化SAR舰船图像检测方法

Lightweight SAR Ship Image Detection Method Based on Improved YOLOv11n

  • 摘要:目的】针对星载与机载等资源受限平台中合成孔径雷达(SAR)舰船检测模型参数量大、计算量高、难以实时部署的问题,对轻量化SAR舰船检测方法进行研究。【方法】以YOLOv11n为基础模型,首先采用引入部分卷积(PConv)重构骨干网络中的C3k2模块,以降低特征提取阶段的计算复杂度;其次设计轻量级分组注意力下采样(LGAD)模块,通过局部自适应注意力机制抑制相干斑噪声干扰,强化对微小舰船目标的特征保留;最后设计轻量化任务对齐检测头(TALD),通过共享卷积架构和任务交互机制解决分类与定位任务的特征不对齐问题。【结果】在SSDD数据集上的实验结果表明,该方法的mAP@0.5达到98.16%,Precision达到97.26%,参数量约为1.2 M,GFLOPs为3.8,FPS达到458。与YOLOv11n相比,参数量减少53.8%,计算量降低39.7%,推理速度提升12.8%。【结论】该方法在保持较高检测精度的同时显著降低了模型复杂度,在SAR舰船检测任务中实现了精度与轻量化的有效平衡,可为资源受限条件下的实时部署提供支持。

     

    Abstract: Objectives To address the problems of large parameter size, high computational cost, and difficulty in real-time deployment of synthetic aperture radar (SAR) ship detection models on resource-constrained platforms such as spaceborne and airborne systems, a lightweight SAR ship detection method was studied. Methods Using YOLOv11n as the baseline model, Partial Convolution (PConv) was first introduced to reconstruct the C3k2 module in the backbone network, so as to reduce the computational complexity during feature extraction. Then, a Lightweight Group-Attention Downsampling (LGAD) module was designed to suppress speckle noise interference through a local adaptive attention mechanism and enhance feature preservation for small ship targets. Finally, a Task-Aligned Lightweight Detection Head (TALD) was developed to alleviate the feature misalignment between classification and localization tasks by using a shared convolution architecture and a task interaction mechanism. Results Experimental results on the SSDD dataset show that the proposed method achieves an mAP@0.5 of 98.16% and a Precision of 97.26%, with about 1.2 M parameters, 3.8 GFLOPs, and an FPS of 458. Compared with YOLOv11n, the number of parameters is reduced by 53.8%, the computational cost is decreased by 39.7%, and the inference speed is improved by 12.8%. Conclusions The proposed method significantly reduces model complexity while maintaining high detection accuracy, achieves an effective balance between accuracy and lightweight design in SAR ship detection, and provides support for real-time deployment under resource-constrained conditions.

     

/

返回文章
返回