• YOLOv8s-SPDX:面向无人机航拍图像的小目标检测算法

    YOLOv8s-SPDX: Small object detection algorithm for UAV aerial images

    • 针对无人机航拍图像中小目标像素占比低、背景复杂以及机载资源受限等问题,本文基于YOLOv8提出一种面向无人机平台的小目标检测改进模型YOLOv8s-SPDX。首先,对基线网络进行轻量级多尺度颈部设计,通过引入专门的小目标检测层增强模型对小目标的检测能力,同时移除冗余的大目标检测层,从而有效降低网络结构的复杂度。其次,在骨干网络和颈部网络中引入空间到深度转换模块(Space-to-Depth,SPD),以消除步长卷积导致的小目标精细特征信息丢失问题。最后,融合C2f模块与轻量级混合局部通道注意力(MixedLocal Channel Attention, MLCA),构建C2f-MLCA模块,使模型能够有效结合局部与全局特征、通道与空间信息,进一步提升复杂背景下的特征表达能力。实验结果表明:在VisDrone-DET2019数据集上,改进模型在验证集上的mAP@0.5从38.6%提升至53.8%,测试集上的mAP@0.5从31.8%提升至42.7%;在TinyPerson和UAVDT-DET数据集上,mAP@0.5分别提升14.5%和2.2%。此外,模型参数量相比基线模型减少20.7%。

       

      Abstract: To address the challenges of low pixel occupancy of small objects, complex backgrounds, and limited onboard resources in UAV aerial images, this paper proposes an improved small object detection model based on YOLOv8, termed YOLOv8s-SPDX. Firstly, a lightweight multi-scale neck design is introduced to the baseline network. A dedicated small object detection layer is incorporated to enhance the model's capability to detect small objects, while redundant large object detection layers are removed, effectively reducing the complexity of the network architecture. Secondly, Space-to-Depth (SPD) modules are integrated into both the backbone and neck networks to eliminate the loss of fine-grained feature information of small objects caused by strided convolutions. Finally, a C2f-MLCA module is constructed by fusing the C2f module with a lightweight Mixed Local Channel Attention (MLCA) mechanism, enabling the model to effectively combine local and global features as well as channel and spatial information, thereby further improving feature representation capability under complex backgrounds. Experimental results demonstrate that on the VisDrone-DET2019 dataset, the proposed model improves mAP@0.5 from 38.6% to 53.8% on the validation set and from 31.8% to 42.7% on the test set. On the TinyPerson and UAVDT-DET datasets, mAP@0.5 is improved by 14.5% and 2.2%, respectively. Furthermore, the number of model parameters is reduced by 20.7% compared to the baseline model.

       

    /

    返回文章
    返回