融合动态特征的无人机小目标检测算法

UAV Small Target Detection Algorithm Integrating Dynamic Features

  • 摘要: 针对无人机航拍图像中目标尺度微小、背景环境复杂以及密集场景下目标分布极不均匀导致的漏检与误检问题,本文提出一种基于改进 YOLOv11 的轻量级小目标检测算法。设计了轻量级动态特征融合模块(LDFM),通过并行的大核深度卷积与几何特征提取分支,结合自适应门控机制,有效增强了微小目标在复杂背景下的特征显著性。其次,提出可学习的密度自适应任务对齐策略(LDA-TAL),利用网格密度估计动态调整正样本分配数量,显著缓解了密集遮挡场景下的样本分配不足问题。最后,引入质量平衡 IoU 损失(QB-IoU),根据预测框的匹配质量对回归损失进行重加权,提升模型对低质量难样本的定位精度。在 VisDrone2019 数据集上的实验结果表明,该算法的平均精度均值(mAP@0.5)达到 46.1%,相比基线模型提升了 11.3%,在保持参数量仅为 2.87\times 10^6 的轻量化优势下,检测精度显著优于现有的主流检测算法。

     

    Abstract: To address the issues of missed and false detections caused by tiny target scales, complex background environments, and extremely uneven target distribution in dense scenes in UAV aerial images, we propose a lightweight small target detection algorithm based on an improved YOLOv11. A light-weight dynamic feature fusion module (LDFM) is designed, which effectively enhances the feature saliency of tiny targets in complex backgrounds through parallel large-kernel depthwise convolutions and geometric feature extraction branches combined with an adaptive gating mechanism. Secondly, a learnable density-adaptive task alignment (LDA-TAL) strategy is proposed, which dynamically adjusts the number of positive samples by estimating grid density, significantly alleviating the problem of insufficient sample allocation in dense occluded scenarios. Finally, a quality-balanced IoU loss (QB-IoU) is introduced, which reweights regression loss according to the matching quality of predicted boxes, improving the localization accuracy of low-quality difficult samples. Experimental results on the VisDrone2019 dataset show that the algorithm achieves an mAP@0.5 of 46.1%, an increase of 11.3% compared with the baseline model, and, while maintaining a lightweight advantage with only 2.87\times 10^6 parameters, the detection accuracy is significantly better than that of existing mainstream detection algorithms.

     

/

返回文章
返回