本文二维码信息
二维码(扫一下试试看!)
Paper Defect Detection Method Based on Improved YOLOv5
Received:May 27, 2022  
DOI:10.11980/j.issn.0254-508X.2022.10.012
Key Words:paper defect detection  YOLOv5  batch normalization  coordinate attention  loss function
Fund Project:陕西省榆林市2020年科技计划项目(CXY-2020-090)。
Author NameAffiliationPostcode
ZHANG Kaisheng* School of Electrical and Control Engineering Shaanxi University of Science & Technology Xi’an Shaanxi Province 710021 710021
GUAN Kaikai School of Electrical and Control Engineering Shaanxi University of Science & Technology Xi’an Shaanxi Province 710021 710021
Hits: 2002
Download times: 1100
Abstract:Aiming at the problems of difficulty in extracting paper defect features and poor real time in current paper defect detection algorithms, an improved YOLOv5 paper defect detection method was proposed in this study. The method was firstly adding centering and scaling calibration to the head and tail of the batch normalization module to form a more stable effective feature distribution of paper defects. Secondly, a coordinate attention was added to the backbone network to enhance the paper defect feature extraction capability of the backbone network. Finally, CIoU_loss was selected as the loss function of bounding box regression to achieve high-precision positioning. The experimental results showed that the average accuracy of the improved algorithm reached 99.02%, the real-time detection speed reached 41.58 frames/s. Compared with the existing CNN-based paper defect classification algorithm, the detection accuracy and detection speed were greatly improved. The improved algorithm was less dependent on the light source and could accurately identify various paper defects.
View Full Text  HTML  View/Add Comment  Download reader