基于空间特征融合下采样卷积的遥感图像目标检测算法OA
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.
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.
林立恒;林珊玲;卢蓓婕;林志贤;郭太良
福州大学 先进制造学院,福建 泉州 362200||中国福建光电信息科学与技术创新实验室,福建 福州 350116福州大学 先进制造学院,福建 泉州 362200||中国福建光电信息科学与技术创新实验室,福建 福州 350116福州大学 先进制造学院,福建 泉州 362200||中国福建光电信息科学与技术创新实验室,福建 福州 350116福州大学 先进制造学院,福建 泉州 362200||中国福建光电信息科学与技术创新实验室,福建 福州 350116||福州大学 物理与信息工程学院,福建 福州 350116中国福建光电信息科学与技术创新实验室,福建 福州 350116||福州大学 物理与信息工程学院,福建 福州 350116
信息技术与安全科学
遥感图像目标检测YOLOv8空间特征融合下采样WIoU
remote sensing image object detectionYOLOv8spatial feature fusiondownsamplingWIoU
《红外技术》 2026 (4)
476-483,8
国家重点研发资助项目(2021YFB3600603),福建省自然科学基金资助项目(2020J01468).
评论