To address the larger numbers of small objects and the issues of occlusion and clustering in UAV aerial photography, which can lead to false positives and missed detections, we propose an improved small object detection algorithm for UAV aerial scenarios called YOLOv8 with tiny prediction head and Space-to-Depth Convolution (HSP-YOLOv8). Firstly, a tiny prediction head specifically for small targets is added to provide higher-resolution feature mapping, enabling better predictions. Secondly, we designed the Space-to-Depth Convolution (SPD-Conv) module to mitigate the loss of small target feature information and enhance the robustness of feature information. Lastly, soft non-maximum suppression (Soft-NMS) is used in the post-processing stage to improve accuracy by significantly reducing false positives in the detection results. In experiments on the Visdrone2019 dataset, the improved algorithm increased the detection precision mAP0.5 and mAP0.5:0.95 values by 11% and 9.8%, respectively, compared to the baseline model YOLOv8s.


    Zugriff

    Download


    Exportieren, teilen und zitieren



    Titel :

    HSP-YOLOv8: UAV Aerial Photography Small Target Detection Algorithm


    Beteiligte:
    Heng Zhang (Autor:in) / Wei Sun (Autor:in) / Changhao Sun (Autor:in) / Ruofei He (Autor:in) / Yumeng Zhang (Autor:in)


    Erscheinungsdatum :

    2024




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Unbekannt




    ITD-YOLOv8: An Infrared Target Detection Model Based on YOLOv8 for Unmanned Aerial Vehicles

    Xiaofeng Zhao / Wenwen Zhang / Hui Zhang et al. | DOAJ | 2024

    Freier Zugriff

    YOLOv8-MPEB small target detection algorithm based on UAV images

    Xu, Wenyuan / Cui, Chuang / Ji, Yongcheng et al. | Elsevier | 2024

    Freier Zugriff

    Improved YOLOv7 Target Detection Algorithm Based on UAV Aerial Photography

    Zhen Bai / Xinbiao Pei / Zheng Qiao et al. | DOAJ | 2024

    Freier Zugriff

    PHSI-RTDETR: A Lightweight Infrared Small Target Detection Algorithm Based on UAV Aerial Photography

    Sen Wang / Huiping Jiang / Zhongjie Li et al. | DOAJ | 2024

    Freier Zugriff

    UAV Detection for Aerial Vehicles using YOLOv8

    Sairam, D.V. / Bhuvaneswari, R / Chokkalingam, SP | IEEE | 2024