重参数化DPE-YOLOv8无人机红外目标检测算法

Re-parameterized DPE-YOLOv8 Unmanned Aerial Vehicle Infrared Target Detection Algorithm

  • 摘要: 针对无人机红外目标检测中存在的小目标细节信息不足、背景复杂及尺度变化大等问题,提出了一种基于改进YOLOv8n的无人机红外目标检测算法DPE-YOLOv8。首先,引入了重参数化C2f_DBB模块,通过多样化分支路径提升模型在复杂背景下对目标的特征提取能力;同时,设计了多尺度自注意力机制PSM,以强化模型对多尺度信息的感知能力;其次,提出了SDC2f模块,在减少计算成本下显著增强网络对小目标的特征表达能力;最后,提出了高效组卷积解耦头Detect_Ghead,利用分组卷积来平衡精度和处理速度。在HIT-UAV数据集上的实验结果表明,DPE-YOLOv8的mAP@50与mAP50:90分别达到了83.4%和53.6%,相较于原始YOLOv8n分别提高了3.4%和1.2%,召回率提升了4.6%,而模型计算复杂度则降低了8.6%。在CTIR数据集上进行泛化测试,进一步验证了本文算法的有效性和鲁棒性。改进后的模型在检测准确性和效率之间实现了良好的平衡,能更好的完成无人机红外目标检测任务。

     

    Abstract: In view of the problems such as complex background interference, weak small targets and multi-scale variations existing in the infrared target detection task of unmanned aerial vehicles, a method named DPE-YOLOv8 based on the improved YOLOv8n is proposed. Firstly, the re-parameterized C2f_DBB module was introduced, which enhances the model's ability to extract features of the target in complex backgrounds through diversified branch paths; simultaneously, the multi-scale self-attention mechanism PSM was designed to strengthen the model's perception of multi-scale information; secondly, the SDC2f module was proposed, which significantly enhances the network's ability to express features of small targets while reducing computational costs. Finally, an efficient group convolution decoupling head structure is proposed in this paper, using group convolution to balance accuracy and processing speed. Experimental results show that the proposed algorithm achieves an mAP@50 of 83.4% on the HIT-UAV dataset, which is 3.4% higher than the original YOLOv8n and 4.6% higher in recall rate, while the model's computational complexity is 8.6% lower. Generalization tests were conducted on the CTIR dataset, further verifying the effectiveness and robustness of the algorithm proposed in this paper. Compared with the existing state-of-the-art methods, the improved model provides a good balance between accuracy and efficiency, fully demonstrating that the improved algorithm outperforms other mainstream algorithms and can better complete the infrared target detection task of UAVs.

     

/

返回文章
返回