• Laser & Optoelectronics Progress
  • Vol. 59, Issue 18, 1815003 (2022)
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
    DOI: 10.3788/LOP202259.1815003 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

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