IEEE Access (Jan 2021)

Research on a Product Quality Monitoring Method Based on Multi Scale PP-YOLO

  • Yiting Li,
  • Haisong Huang,
  • Qipeng Chen,
  • Qingsong Fan,
  • Huafeng Quan

DOI
https://doi.org/10.1109/ACCESS.2021.3085338
Journal volume & issue
Vol. 9
pp. 80373 – 80387

Abstract

Read online

To monitor product quality in the production process in real time, this thesis proposes a quality monitoring model based on PaddlePaddle You Only Look Once (PP-YOLO). First, in the preprocessing stage, the data enhancement method and the K-means++ method are used to improve the robustness of the algorithm, and the generated anchor box can screen more refined features earlier. Second, ResNet50-vd with the deformable convolution idea is selected as the backbone of the detection model, the feature pyramid network structure and the composition of the loss function are improved, and the feature learning ability of the model is enhanced to enable it to detect multiple scales of defects. Finally, pruning is performed on the basis of the trained model to reduce the number of model parameters so that it can be deployed in industrial scenarios with limited hardware conditions. Experimental results show that the proposed quality monitoring model can meet the requirements for detection speed and accuracy in actual production, providing a new concept for the deployment of deep learning models in the industrial field.

Keywords