• 基于改进YOLOv8s的航拍图像小目标检测算法

    Small target detection algorithm for aerial images based on improved YOLOv8s

    • 针对无人机图像中小目标检测存在的目标尺度变化大、漏检、误检和目标遮挡等问题,提出了一种改进的SDP-YOLOv8s算法。该算法首先在检测头部分引入小目标检测头并移除大目标检测头,在提升小目标检精度的同时减少算法冗余计算量。其次,在特征融合层中引入并行位置感知注意力模块PPA,通过捕捉多尺度特征信息提高特征融合能力。设计动态上采样模块X-DySample,进一步优化算法处理不同尺度特征的能力,提升算法的抗干扰能力;并在主干网络中引入SPD-Conv模块,改善卷积过程中特征丢失问题。在公开数据集VisDrone2019上的实验结果表明:相较于YOLOv8s算法,SDP-YOLOv8算法在mAP@0.5和mAP@0.5:0.95上的检测精度分别提升了8.2%和5.8%,算法参数量降低了22.5%。同时,在Tiny-Person数据集上验证了所提出算法的泛化性和有效性。

       

      Abstract: For the problems such as large-scale target variation, missed detection, false detection and target occlusion in the detection of small targets in drone images, an improved SDP-YOLOv8s algorithm is proposed. Firstly, a small target detection head is introduced in the detection head part and the large target detection head is removed, which improves the accuracy of small target detection while reducing the redundant computational load of the algorithm. Secondly, a parallel position-aware attention module (PPA) is introduced in the feature fusion layer to capture multi-scale feature information and enhance the feature fusion ability. A dynamic upsampling module X-DySample is designed to further optimize the algorithm's ability to process features of different scales and improve the algorithm's anti-interference ability, and the SPD-Conv module is introduced in the backbone network to improve the problem of feature loss during convolution. Experimental results on the public dataset VisDrone2019 show that compared with the YOLOv8s algorithm, the SDP-YOLOv8 algorithm has improved the detection accuracy by 8.2% and 5.8% in mAP@0.5 and mAP@0.5:0.95 respectively, and the algorithm parameters have been reduced by 22.5%. At the same time, the generalization and effectiveness of the proposed algorithm have been verified on the Tiny-Person dataset.

       

    /

    返回文章
    返回