基于小波浅层残差网络的SAR图像船舰小目标检测方法

    A Ship Small-Target Detection Method in SAR Images Based on Wavelet Shallow Residual Network

    • 在合成孔径雷达(Synthetic Aperture Radar,SAR)图像船舰小目标检测任务中,由噪声、杂波干扰等因素造成的细节信息丢失和边缘模糊会导致检测精度下降。针对此问题,文章提出了一种基于小波浅层残差网络(Wavelet Shallow Skip Residual Network,WSSRNet)的SAR图像船舰小目标检测方法。首先,在主干特征提取网络中构建多尺度小波卷积残差模块(Multi-scale Wavelet Transform Residual Module,MWTRM),引入小波变换卷积获取丰富的感受野并融合多尺度特征,以增强对上下文以及空间特征信息的关注,在降低背景干扰的同时减少计算开销。其次,提出浅层跳跃残差模块(Shallow Skip Residual Module,SSRM),通过设计浅层特征金字塔结构与跨阶段特征融合,增强对位置信息及浅层细节特征的利用,提升小目标的定位精度与召回率。最后,提出了一种新的损失函数Inner-ShapeIoU,根据目标的尺度差异自适应调整回归策略,提高网络的泛化性能以及对目标的定位精度。在HRSID数据集上的实验结果表明,文章方法准确率达92.3%,召回率达87.1%,平均检测精度(mean Average Precision,mAP)在IoU阈值为0.5的条件下达94.7%,在0.5~0.95的条件下达70.2%。文章方法在检测精度、参数量及对小目标特征提取能力上表现卓越,多场景检测性能强、鲁棒性佳,能显著改善小目标在复杂场景下的错检、漏检问题。

       

      Abstract: In Synthetic Aperture Radar (SAR) image ship detection, the loss of detailed information and edge blurring caused by coherent speckle noise and low signal-to-clutter ratio often degrade detection accuracy. To address this problem, this paper proposes a ship detection algorithm for SAR images based on the Wavelet Shallow Residual Network (WSSRNet). First, a Multi-scale Wavelet Transform Residual Module (MWTRM) is constructed in the backbone feature extraction network, which introduces wavelet transform convolution to capture diverse receptive fields and fuse multi-scale features, thereby enhancing the attention to contextual and spatial feature information while reducing background interference and computational overhead. Second, a Shallow Skip Residual Module (SSRM) is proposed, which enhances the utilization of positional information and shallow fine-grained features by designing a shallow feature pyramid structure and cross-stage feature fusion, improving the boundary localization precision and recall of small targets. Finally, a novel loss function, Inner-ShapeIoU, is introduced, which adaptively adjusts the regression strategy based on the scale differences of targets, improving the network's generalization performance and boundary localization precision. Experimental results on the High-Resolution SAR Images for Ship Detection (HRSID) dataset show that the proposed method achieves an accuracy of 92.3%, a recall of 87.1%, and a mean average precision (mAP) of 94.7% at an IoU threshold of 0.5, and 70.2% at IoU thresholds of 0.5-0.95. The proposed method demonstrates superior performance in detection accuracy, parameter efficiency, and small target feature extraction capability, with strong multi-scene detection performance and robustness, significantly alleviating the problems of false detection and missed detection for small targets in complex scenarios.

       

    /

    返回文章
    返回