基于空间特征融合下采样卷积的遥感图像目标检测算法

Remote Sensing Image Target Detection Algorithm Based on Spatial Feature Fusion Down-sampling Convolution

  • 摘要: 针对遥感图像中小目标繁多,现有目标检测算法在遥感图像目标检测任务中性能低下,小目标检测困难等问题,提出一种基于空间特征融合下采样卷积的遥感图像目标检测算法。首先,为避免特征图经过现有下采样卷积模块造成的小目标信息丢失,设计了一种空间特征融合下采样卷积模块代替传统下采样卷积,使模型在训练和推理过程中更聚焦于小目标。其次,引入轻量化颈部特征融合模块,并在主干网络部分搭配无参注意力机制模块,在保证模型精度的同时降低模型参数量和复杂度。最后,采用WIoU作为回归损失,提高模型回归框精度。实验证明,相比YOLOv8n,在DIOR遥感图像目标检测数据集中mAP50提高了2.3%,mAP50-90提高了2.1%,且参数量下降了0.12 M。

     

    Abstract: In view of the large number of small targets in remote sensing images, the limited performance of existing detection algorithms, and the inherent difficulty of small-target detection, this study proposes a novel target detection algorithm based on spatial feature fusion down-sampling convolution. First, to mitigate the loss of small-target information caused by conventional down-sampling convolution modules, a spatial feature fusion down-sampling convolution module is designed to replace traditional down-sampling convolution, enabling the model to better focus on small targets during both training and inference. Second, a lightweight neck feature fusion module is introduced, incorporating a parameter-free attention mechanism to maintain detection accuracy while reducing model complexity and parameter count. Finally, WIoU is employed as a regression loss function to improve the accuracy of the regression frame. Experimental results on the DIOR remote sensing image dataset show that, compared with YOLOv8n, the proposed method achieves improvements of 2.3% in mAP50 and 2.1% in mAP50-90 while reducing the number of parameters by 0.12 M.

     

/

返回文章
返回