Electronics Optics & Control, Volume. 31, Issue 12, 8(2024)

Improved YOLOv7 Based UAV Image Small Target Detection Algorithm

WANG Xiaoyu1... ZHANG Lihui1,2, ZHAO Hui1, and ZHANG Lijuan13 |Show fewer author(s)
Author Affiliations
  • 1College of Computer Science and Engineering, Changchun University of Technology, Changchun 130000, China
  • 2School of Management Engineering, Jilin University of Architecture and Technology, Changchun 130000, China
  • 3School of Internet of Things Engineering, Wuxi University, Wuxi 214000, China
  • show less

    Aiming at the problems of low accuracy, missing detection and false detection in the identification of small targets in UAV aerial images by existing target detection methods, a small target detection method based on improved YOLOv7 is proposed. Firstly, the network detection head is improved, the 20×20 detection head originally used for detecting large objects is removed, a 160×160 detection head for small targets is added, and the network feature fusion path is modified accordingly. Secondly, the SPPCSPC structure is reconstructed, the convolutional layer is clipped and the pooling structure is changed to reduce module complexity and speed up network convergence. Then, the original upsampling structure is replaced by the Content-Aware ReAssembly of FEatures (CARAFE) operator to reduce the loss of image information during upsampling and maximize the preservation of local and corner information of the input image. Finally, the ELAN module is improved to lighten the backbone while improving the sensitivity of the network to small-scale targets. Experiments are carried out on the public dataset VisDrone2019, and the mAP50 of the improved model reached 56.6%, which is 2.9 percentage points higher than that of the original YOLOv7 model, and the parameters are reduced by 33%.

    Tools

    Get Citation

    Copy Citation Text

    WANG Xiaoyu, ZHANG Lihui, ZHAO Hui, ZHANG Lijuan. Improved YOLOv7 Based UAV Image Small Target Detection Algorithm[J]. Electronics Optics & Control, 2024, 31(12): 8

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Dec. 5, 2023

    Accepted: Dec. 25, 2024

    Published Online: Dec. 25, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.12.002

    Topics