Laser & Optoelectronics Progress, Volume. 57, Issue 16, 161001(2020)
Real-time Fabric Defect Detection Algorithm Based on S-YOLOV3 Model
To meet the real-time requirements of fabric defect detection in industry, a real-time fabric defect detection algorithm based on S-YOLOV3 (Slimming You Only Look Once Version 3) model is proposed. To develop this algorithm, the K-means clustering algorithm is used to determine the target prior frame for adapting to different sizes of defects. The YOLOV3 model is then pretrained to obtain the weight parameters, and the scaling factor γ is used in the batch normalization layer to evaluate the weight of each convolution kernel. The convolution kernel with weight value lower than the threshold is clipped to obtain the S-YOLOV3 model to achieve compression and acceleration. Finally, the pruned network is fine-tuned to improve the model detection accuracy. Experiment results reveal that the proposed model provides accurate detection of fabrics with different complex textures (average precision of 94%). After pruning, the detection speed is increased to 55 FPS. The obtained accuracy and real-time can meet the actual demand of industry.
Get Citation
Copy Citation Text
Jun Zhou, Junfeng Jing, Huanhuan Zhang, Zhen Wang, Hanlin Huang. Real-time Fabric Defect Detection Algorithm Based on S-YOLOV3 Model[J]. Laser & Optoelectronics Progress, 2020, 57(16): 161001
Category: Image Processing
Received: Nov. 18, 2019
Accepted: Dec. 31, 2019
Published Online: Aug. 5, 2020
The Author Email: Jing Junfeng (1302230897@qq.com)