In this paper, an object detection and recognition method based on improved YOLOv5 is proposed for application on unmanned aerial vehicle (UAV) aerial images. Firstly, we improved the traditional Gabor function to obtain Gabor convolutional kernels with better edge enhancement properties. We used eight Gabor convolutional kernels to enhance the object edges from eight directions, and the enhanced image has obvious edge features, thus providing the best object area for subsequent deep feature extraction work. Secondly, we added a coordinate attention (CA) mechanism to the backbone of YOLOv5. The plug-and-play lightweight CA mechanism considers information of both the spatial location and channel of features and can accurately capture the long-range dependencies of positions. CA is like the eyes of YOLOv5, making it easier for the network to find the region of interest (ROI). Once again, we replaced the Path Aggregation Network (PANet) with a Bidirectional Feature Pyramid Network (BiFPN) at the neck of YOLOv5. BiFPN performs weighting operations on different input feature layers, which helps to balance the contribution of each layer. In addition, BiFPN adds horizontally connected feature branches across nodes on a bidirectional feature fusion structure to fuse more in-depth feature information. Finally, we trained the overall improved YOLOv5 model on our integrated dataset LSDUVD and compared it with other models on multiple datasets. The results show that our method has the best convergence effect and mAP value, which demonstrates that our method has unique advantages in processing detection tasks of UAV aerial images.


    Access

    Download


    Export, share and cite



    Title :

    Research on Object Detection and Recognition Method for UAV Aerial Images Based on Improved YOLOv5


    Contributors:
    Heng Zhang (author) / Faming Shao (author) / Xiaohui He (author) / Zihan Zhang (author) / Yonggen Cai (author) / Shaohua Bi (author)


    Publication date :

    2023




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




    Research on Improved YOLOv5 Vehicle Target Detection Algorithm in Aerial Images

    Xue Yang / Jihong Xiu / Xiaojia Liu | DOAJ | 2024

    Free access

    An improved YOLOv5 method for small object detection in high resolution images

    Ran, Dongni / Xiong, Xuhui / Gao, Lujunjie | SPIE | 2023


    YOLOv5-Based Object Detection for Emergency Response in Aerial Imagery

    Boddu, Sindhu / Mukherjee, Arindam | ArXiv | 2024

    Free access

    An Improved YOLOv5-Based Small Target Detection Method for UAV Aerial Image

    Li, Ruoyu / Gao, Yang / Zhang, Ruixing | Springer Verlag | 2023


    Research on ship detection technology based on improved YOLOv5

    Huan, Yutai / Chen, Lin / Liu, Bin et al. | IEEE | 2023