The improvement of small target detection ability determines the effect of target recognition in the field of target detection. To solve the problems of inaccurate positioning of small target detection, difficulty in feature extraction, and easy-to-miss detection, a small vehicle target detection algorithm based on YOLOv4 was proposed in this paper, namely RC_IYOLOv4, based on VEDAI, a public data set of small vehicle target under UAV. Firstly, the k-Means ++ clustering algorithm retrieves the anchor point box conforming to the VEDAI data set. Then, feature extraction was enhanced by embedding RFB receptive field module and amplifying small target receptive fields by replacing the SPP module with RFB. Finally, this paper adds the CBAM attention mechanism module. It uses one-dimensional convolution to replace the improved combination of complete connection and parallel connection in its interior with focusing on learning the feature representation of small targets. Based on the improved methods above, the experimental results show that the proposed improved algorithm RC_IYOLOv4 improves 4.98% in the VEDAI data set MAP@0.5 compared with YOLOv4. Finally, in the PASCAL VOC data set commonly used for target detection, MAP@0.5 improves by 2.27%.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Research of Vehicle Detection under UAV Based on YOLOv4 Improved Algorithm


    Contributors:
    Sun, Xueqing (author) / Li, Yuhan (author) / Zhou, Zhiguo (author)


    Publication date :

    2022-10-12


    Size :

    1720568 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    Pedestrian Detection Algorithm Based on Improved YOLOv4

    Zheng, Ziheng / Ni, Chenhui / Zeng, Guolei | IEEE | 2023


    Detection Algorithm Based on Improved YOLOv4 for Pedestrian

    Kong, Wen / Qiao, Yidan / Wei, Ziqi | IEEE | 2022


    UAV Target Detection Algorithm Based on Improved YOLOv4

    Wang, Wenyue / Li, Jian / Zhang, Qi | Springer Verlag | 2022


    Maritime ship detection algorithm based on improved YOLOv4

    Chen, Wangcheng / Li, Yingshu / Wang, Xuemei et al. | SPIE | 2023


    A Improved Yolov4’s vehicle and pedestrian detection method

    Wang, Hailong / Tian, Shihe / Zhang, Zhian et al. | VDE-Verlag | 2022