基于RSC-YOLO的无人机目标检测算法

UAV Object Detection Algorithm Based on RSC-YOLO

  • 摘要: 针对基于无人机平台的目标检测中存在的背景复杂、目标密集、相互遮挡等问题,提出了一种基于YOLOv11n的改进算法RSC-YOLO。首先,引入感受野注意力机制RFA,重新设计Conv和C3k2,有效解决了空间特征重叠导致的参数共享问题;其次,针对颈部设计了一种浅层细节融合模块SDFM,强调对浅层特征的关注度,补偿小目标的特征缺失以及维护遮挡目标剩余空间信息的完整性;最后,在颈部引入了轻量级跨尺度特征融合网络CCFF,增强了模型的特征交互能力和多尺度特征融合能力,实现了精度与检测速度之间的平衡。在Visdrone2019数据集上进行消融实验和对比实验,mAP50相较基线算法提升了4.5%,FPS达到213,能够满足实时性的检测需求;在DOTA数据集和HIT-UAV红外小目标数据集上进行泛化实验,mAP50分别提升了3.1%、3.9%,证明所提算法具有广泛适用性。

     

    Abstract: An improved algorithm called RSC-YOLO based on YOLOv11n is proposed to address the problems of complex backgrounds, dense objects, and mutual occlusion in object detection based on uncrewed aerial vehicle (UAV) platforms. Firstly, the representative feature alignment (RFA) sensing field attention mechanism is introduced, and the Conv and C3k2 models are re-designed, which effectively solves the parameter sharing problem caused by the overlapping of spatial features. Secondly, a shallow detail fusion module(SDFM) is designed for the neck network of YOLO, which emphasizes the attention to shallow features, compensates for the missing features of small objects as well as maintains the integrity of the remaining spatial information of occluded objects. Finally, a lightweight cross-scale feature fusion network CCFF is introduced for the neck, which enhanced the model's feature interaction capability and multi-scale feature fusion capability to achieve a balance between accuracy and detection speed. Ablation and comparison experiments were conducted on the Visdrone2019 dataset, and the mAP50 was improved by 4.5% compared with the baseline algorithm. The framerate of the system reached 213 FPS, which is sufficient for real-time detection. Generalization experiments were conducted on the DOTA dataset and the HIT-UAV infrared small object dataset, and the results show that the mAP50 was improved by 3.1% and 3.9%, respectively, which demonstrates that the proposed algorithm has wide applicability.

     

/

返回文章
返回