• Electronics Optics & Control
  • Vol. 30, Issue 2, 24 (2023)
YANG Jinhui1、2、3, LI Hong1、2、3, DU Yunyan1、2、3, MAO Yao1、2、3, and LIU Qiong1、2、3
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • show less
    DOI: 10.3969/j.issn.1671-637x.2023.02.005 Cite this Article
    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 Copy Citation Text show less

    Abstract

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