• 基于多尺度融合与小目标增强的航拍车辆检测算法

    Aerial vehicle detection algorithm based on multi-scale fusion and small-object enhancement

    • 针对无人机视角下车辆目标检测精度低、漏检与误检率高的问题,提出了一种基于YOLOv11n的车辆检测模型YOLO-VC(YOLO-Vehicle Capture)。首先,在主干网络中提出PMSFA(部分多尺度特征提取模块)代替C3k2模块,利用其对不同尺度特征的差异化融合策略,增强多尺度特征提取能力;其次,在颈部网络中引入PSConv(风车卷积),利用其不对称的卷积核,在复杂背景噪声中增强对微小特征的捕获;然后,提出小目标特征增强金字塔,将P2层特征与P3层特征融合,并由DBIE(双分支交互增强模块)进行特征增强,提高小目标检测精度;最后,采用PIoU V2(Powerful-IoUV2)损失函数,增强对中等质量锚框的聚焦,提高对密集目标的检测精度。实验结果表明,在VisDrone2019数据集上,本文算法的mAP@50相较基线模型提高了6.2%,模型大小仅为3.72 MB。

       

      Abstract: To address the challenges of low detection accuracy, high miss rates, and elevated false-positive rates in vehicle detection from UAV perspectives, this paper proposes a vehicle detection model termed YOLO-VC (YOLO-Vehicle Capture), built upon YOLOv11n. First, a Partial Multi-Scale Feature Aggregation (PMSFA) module is introduced in the backbone network to replace the original C3k2 module, leveraging a differentiated fusion strategy across scales to enhance multi-scale feature extraction capability. Second, PSConv (Pyramid-Structured Convolution) is incorporated into the neck network, whose asymmetric convolutional kernels facilitate the capture of subtle features amid complex background clutter. Third, a small-object feature enhancement pyramid is constructed by fusing P2-level and P3-level features, which are subsequently refined by a Dual-Branch Interactive Enhancement (DBIE) module to boost small-object detection accuracy. Finally, the PIoU V2 (Powerful-IoU Version 2) loss function is adopted to strengthen the focus on medium-quality anchor boxes, thereby improving detection precision in dense scenes. Experimental results on the VisDrone2019 dataset demonstrate that the proposed algorithm achieves a 6.2 percentage-point improvement in mAP@50 over the baseline model, while maintaining a compact model size of only 3.72 MB.

       

    /

    返回文章
    返回