Affiliation:
1. School of Computer Science, Xi’an Polytechnic University, Xi’an 710048, China
2. Shaanxi Key Laboratory of Clothing Intelligence, School of Computer Science, Xi’an Polytechnic University, Xi’an 710048, China
Abstract
Defect detection holds significant importance in improving the overall quality of fabric manufacturing. To improve the effectiveness and accuracy of fabric defect detection, we propose the PRC-Light YOLO model for fabric defect detection and establish a detection system. Firstly, we have improved YOLOv7 by integrating new convolution operators into the Extended-Efficient Layer Aggregation Network for optimized feature extraction, reducing computations while capturing spatial features effectively. Secondly, to enhance the performance of the feature fusion network, we use Receptive Field Block as the feature pyramid of YOLOv7 and introduce Content-Aware ReAssembly of FEatures as upsampling operators for PRC-Light YOLO. By generating real-time adaptive convolution kernels, this module extends the receptive field, thereby gathering vital information from contexts with richer content. To further optimize the efficiency of model training, we apply the HardSwish activation function. Additionally, the bounding box loss function adopts the Wise-IOU v3, which incorporates a dynamic non-monotonic focusing mechanism that mitigates adverse gradients from low-quality instances. Finally, in order to enhance the PRC-Light YOLO model’s generalization ability, we apply data augmentation techniques to the fabric dataset. In comparison to the YOLOv7 model, multiple experiments indicate that our proposed fabric defect detection model exhibits a decrease of 18.03% in model parameters and 20.53% in computational load. At the same time, it has a notable 7.6% improvement in mAP.
Funder
National Natural Science Foundation of China
Subject
Fluid Flow and Transfer Processes,Computer Science Applications,Process Chemistry and Technology,General Engineering,Instrumentation,General Materials Science
Reference40 articles.
1. Attention-based Feature Fusion Generative Adversarial Network for yarn-dyed fabric defect detection;Zhang;Text. Res. J.,2023
2. Li, W., Zhang, Z., Wang, M., and Chen, H. (2023). Fabric Defect Detection Algorithm Based on Image Saliency Region and Similarity Location. Electronics, 12.
3. Yarn-dyed fabric defect detection based on an improved autoencoder with Fourier convolution;Xiang;Text. Res. J.,2023
4. CARL-YOLOF: A well-efficient model for digital printing fabric defect detection;Wu;J. Eng. Fibers Fabr.,2022
5. Saliency-based fabric defect detection via bag-of-words model;Kanwal;Signal Image Video Process.,2023
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献