Electronics Optics & Control, Volume. 30, Issue 2, 24(2023)

A Lightweight Object Detection Algorithm Based on Improved YOLOv5s

YANG Jinhui1...2,3, LI Hong1,2,3, DU Yunyan1,2,3, MAO Yao1,2,3, and LIU Qiong1,23 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • show less

    Aiming at the insufficient feature extraction of the neck feature extraction network PANET of current YOLOv5s, and the conventional convolution Conv consumes a large amount of parameters and calculations, a lightweight target detection algorithm (RFBG-YOLO) is proposed.Firstly, in order to improve the recognition effect of the detector, a multi-branch atrous convolution structure RFB-Bottleneck is proposed to improve the feature extraction ability of PANET, thereby improving the detection accuracy of the model.Then, in order to make the model more lightweight, GhostConv convolution is introduced to reduce the amount of model parameters and improve the detection speed.The results on the PASCAL VOC data set show that in the case of small impact on the detection speed, the mAP@0.5 of the RFBG-YOLO algorithm is 80.3%, an increase of 2.2 percentage points compared with that of YOLOv5s algorithm; mAP@0.5∶0.95 is 55.1%, an increase of 4.2 percentage points compared with that of YOLOv5s algorithm; the amount of model parameters is 5.2 MiB, a reduction of 2.0 MiB compared with that of YOLOv5s algorithm.Therefore, the proposed RFBG-YOLO algorithm has a high enough detection accuracy while ensuring the light weight of the model, which can meet the requirements of detection accuracy in the scenario of lightweight target detection.

    Tools

    Get Citation

    Copy Citation Text

    YANG Jinhui, LI Hong, DU Yunyan, MAO Yao, LIU Qiong. A Lightweight Object Detection Algorithm Based on Improved YOLOv5s[J]. Electronics Optics & Control, 2023, 30(2): 24

    Download Citation

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

    Category:

    Received: Dec. 21, 2021

    Accepted: --

    Published Online: Apr. 3, 2023

    The Author Email:

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

    Topics