Real-time Fabric Defect Detection Algorithm Based on S-YOLOV3 Model

被引:11
作者
Zhou Jun [1 ,2 ]
Jing Junfeng [1 ]
Zhang Huanhuan [1 ]
Wang Zhen [1 ]
Huang Hanlin [1 ]
机构
[1] Xian Polytech Univ, Sch Elect Informat, Xian 710018, Shaanxi, Peoples R China
[2] Xian Polytech Univ, Collaborat Innovat Ctr, Xian 710018, Shaanxi, Peoples R China
关键词
image processing; fabric defects; S-YOLOV3; K-means; model pruning; defect detection;
D O I
10.3788/LOP57.161001
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
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 gamma 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.
引用
收藏
页数:9
相关论文
共 17 条
[1]  
[Anonymous], 2018, ACTA OPTICA SINICA, DOI DOI 10.1042/BSR20180244
[2]  
He K., 2016, Proceedings of the IEEE conference on computer vision and pattern recognition, DOI DOI 10.1109/CVPR.2016.90
[3]  
Jing JF, 2017, TEXT BIOENG INFORM S, P1030
[4]  
[景军锋 Jing Junfeng], 2017, [纺织学报, Journal of Textile Research], V38, P68
[5]   Improved YOLO V3 Algorithm and Its Application in Small Target Detection [J].
Ju Moran ;
Luo Haibo ;
Wang Zhongbo ;
He Miao ;
Chang Zheng ;
Hui Bin .
ACTA OPTICA SINICA, 2019, 39 (07)
[6]   A universal defect detection approach for various types of fabrics based on the Elo-rating algorithm of the integral image [J].
Kang, Xuejuan ;
Zhang, Erhu .
TEXTILE RESEARCH JOURNAL, 2019, 89 (21-22) :4766-4793
[7]   A defect detection method for unpatterned fabric based on multidirectional binary patterns and the gray-level co-occurrence matrix [J].
Li, Feng ;
Yuan, Lina ;
Zhang, Kun ;
Li, Wenqing .
TEXTILE RESEARCH JOURNAL, 2020, 90 (7-8) :776-796
[8]  
Li M, 2018, J XIAN POLYTECHNIC U, V32, P663
[9]   Textile fabric defect detection based on low-rank representation [J].
Li, Peng ;
Liang, Junli ;
Shen, Xubang ;
Zhao, Minghua ;
Sui, Liansheng .
MULTIMEDIA TOOLS AND APPLICATIONS, 2019, 78 (01) :99-124
[10]   Automatic fabric defect detection with a wide-and-compact network [J].
Li, Yuyuan ;
Zhang, Dong ;
Lee, Dah-Jye .
NEUROCOMPUTING, 2019, 329 :329-338