• Electronics Optics & Control
  • Vol. 31, Issue 12, 8 (2024)
WANG Xiaoyu1, ZHANG Lihui1,2, ZHAO Hui1, and ZHANG Lijuan1,3
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
    DOI: 10.3969/j.issn.1671-637x.2024.12.002 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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%.