Electronics Optics & Control, Volume. 29, Issue 5, 23(2022)

Multi-target Detection Method of Aerial Images Based on Improved YOLOv4 Algorithm

WANG Haoxue1... CAO Jie2, QIU Cheng1 and LIU Yaohui1 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    With the gradual maturity of the target detection algorithm based on deep learning, its deployment on UAV has become a hot topic.Regarding the problems of low detection accuracy caused by many small targets and easy occlusion in UAV aerial images, complex detection scenes and large scale variability, an improved algorithm based on the proposed S-YOLOv4 is introduced.Firstly, SENet attention mechanism is added on the original feature extraction network structure to improve the models ability to concentrate on useful information and enhance inter-channel attention.Secondly, a new detection layer with a resolution of 160×160 is added to refine the grid for better detection of small targets.Finally, the loss function is improved, and the class smoothing label is applied to the classification loss to reduce the penalty of negative samples and improve the generalization ability of the model.Compared with that of the original algorithm, the mAP of the proposed algorithm is improved by 3.4% under the real-time detection speed.

    Tools

    Get Citation

    Copy Citation Text

    WANG Haoxue, CAO Jie, QIU Cheng, LIU Yaohui. Multi-target Detection Method of Aerial Images Based on Improved YOLOv4 Algorithm[J]. Electronics Optics & Control, 2022, 29(5): 23

    Download Citation

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

    Category:

    Received: Apr. 9, 2021

    Accepted: --

    Published Online: Aug. 1, 2022

    The Author Email:

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

    Topics