熵引导动态损失加权的SAR图像舰船检测*

    • 摘要: 针对合成孔径雷达(SAR)图像舰船目标检测任务中因噪声和近岸复杂背景干扰导致的漏检、误检率高,以及模型复杂度高、检测实时性不足等问题,本文提出一种熵引导动态损失加权的SAR图像舰船检测方法。首先,利用图像熵量化噪声与近岸背景的干扰程度,动态调整损失函数权重以强化对干扰严重舰船目标的特征学习;然后,以YOLOv8n为基线模型,在其多尺度特征融合模块嵌入高效金字塔池化通道注意力机制(EPPCA),实现跨尺度特征通道的自适应加权融合;最后,通过Grad-CAM技术分析网络层对舰船特征的关注度,剔除冗余层以降低模型复杂度。在SSDD数据集上的实验表明,与原始YOLOv8n相比,该方法模型参数量减少至32.6%,计算量降低至76.5%,精确率、召回率、平均精度均值(mAP50和mAP50-95)分别提升2.4%、1%、0.9%和1.5%,在降低模型复杂度的同时显著提升了检测精度。

       

      Abstract: Aiming at the problems such as high missed detection and false detection rates caused by noise and complex nearshore background interference in the ship target detection task of synthetic aperture radar (SAR) images, as well as high model complexity and insufficient real-time detection performance, this paper proposes a method of Ship detection in SAR images based on entropy-guided loss weighted. Firstly, the interference degree between noise and the nearshore background is quantified by using image entropy, and the weight of the loss function is dynamically adjusted to enhance the feature learning of ship targets with severe interference. Then, taking YOLOv8n as the baseline model, an Efficient Pyramid Pooling Channel Attention mechanism (EPPCA) is embedded in its multi-scale feature fusion module to achieve adaptive weighted fusion of cross-scale feature channels; Finally, the attention of the network layer to the characteristics of ships is analyzed through the Grad-CAM technology, and the redundant layers are eliminated to reduce the complexity of the model. Experiments on the SSDD (SAR Ship Detection Dataset) dataset show that, compared with the original YOLOv8n, the number of model parameters of this method is reduced to 32.6% and the computational cost is reduced to 76.5%. The Precision, Recall, and mean average precision (mAP50 and MAP50-95) increased by 2.4%, 1%, 0.9%, and 1.5% respectively, significantly improving the detection accuracy while reducing the complexity of the model.

       

    /

    返回文章
    返回