Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1815003(2022)

Indoor Scene Object Detection Based on Improved YOLOv4 Algorithm

Weigang Li*, Chao Yang, Lin Jiang, and Yuntao Zhao
Author Affiliations
  • Engineering Research Center for Metallurgical Automation and Measurement Technology, Ministry of Education, Wuhan University of Science and Technology, Wuhan 430081, Hubei , China
  • show less

    In this paper, we proposed an improved YOLOv4 algorithm model to solve the problems of low detection accuracy and slow detection speed of traditional indoor scene object detection methods. First, we constructed an indoor scene object detection dataset. Then, we applied the K-means++ clustering algorithm to optimize the parameters of the priori box and improve the matching degree between the priori box and object. Next, we adjusted the network structure of the original YOLOv4 model and integrated the cross stage partial network architecture into the neck network of the model. This eliminates the gradient information redundancy phenomenon caused by the gradient backpropagation in the feature fusion stage and improves the detection ability for indoor targets. Furthermore, we introduced a depthwise separable convolution module to replace the original 3×3 convolution layer in the model to reduce the model parameters and improve the detection speed. The experimental results show that the improved YOLOv4 algorithm achieves an average accuracy of 83.0% and a detection speed of 72.1 frame/s on the indoor scene target detection dataset, which is 3.2 percentage points and 6 frame/s higher than the original YOLOv4 algorithm, respectively, additionally, the model size is reduced by 36.3%. The improved YOLOv4 algorithm outperforms other indoor scene object detection algorithms based on deep learning.

    Tools

    Get Citation

    Copy Citation Text

    Weigang Li, Chao Yang, Lin Jiang, Yuntao Zhao. Indoor Scene Object Detection Based on Improved YOLOv4 Algorithm[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1815003

    Download Citation

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

    Category: Machine Vision

    Received: Jun. 18, 2021

    Accepted: Jul. 20, 2021

    Published Online: Aug. 29, 2022

    The Author Email: Li Weigang (liweigang.luck@foxmail.com)

    DOI:10.3788/LOP202259.1815003

    Topics