Author:
Lin Guijuan,Liu Keyu,Xia Xuke,Yan Ruopeng
Abstract
Limited by computing resources of embedded devices, there are problems in the field of fabric defect detection, including small defect size, extremely unbalanced aspect ratio of defect size, and slow detection speed. To address these problems, a sliding window multihead self-attention mechanism is proposed for the detection of small targets, and the Swin Transformer module is introduced to replace the main module in the original YOLOv5 algorithm. First, to reduce the distance between several scales, the weighted bidirectional feature network is employed on embedded devices. In addition, it is helpful to improve the perception of small-target faults by incorporating a detection layer to achieve four-scale detection. At last, to improve the learning of positive sample instances and lower the missed detection rate, the generalized focal loss function is finally implemented on YOLOv5. Experimental results show that the accuracy of the improved algorithm on the fabric dataset reaches 85.6%, and the mAP is increased by 4.2% to 76.5%, which meets the requirements for real-time detection on embedded devices.
Funder
the Natural Science Foundation of China
the Natural Science Foundation of Fujian Province, China
Subject
Electrical and Electronic Engineering,Biochemistry,Instrumentation,Atomic and Molecular Physics, and Optics,Analytical Chemistry
Reference39 articles.
1. Research progress of image processing technology for fabric defect detection;Lu;Fangzhi Xuebao/J. Text. Res.,2021
2. Gustian, D.A., Rohmah, N.L., Shidik, G.F., Fanani, A.Z., Pramunendar, R.A. (2019, January 21–22). Classification of Troso Fabric Using SVM-RBF Multi-class Method with GLCM and PCA Feature Extraction. In Proceedings of the 2019 International Seminar on Application for Technology of Information and Communication, iSemantic 2019, Semarang, Indonesia.
3. Defect Detection for Patterned Fabric Images Based on GHOG and Low-Rank Decomposition;Li;IEEE Access,2019
4. Pan, Z., He, N., and Jiao, Z. (2017, January 25–26). FFT used for fabric defect detection based on CUDA. Proceedings of the 2nd IEEE Advanced Information Technology, Electronic and Automation Control Conference, IAEAC 2017, Chongqing, China.
5. Fabric Defects Detection using Adaptive Wavelets;Wen;Int. J. Cloth. Sci. Technol.,2014
Cited by
21 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献