Abstract
In the textile industry, detecting defects in fabrics featuring intricate patterns and small imperfections poses a persistent challenge. There is an urgent need for defect detection systems characterized by high accuracy and ease of deployment. To address these needs, a fabric defect detection algorithm, SR-NET, has been proposed in this paper, featuring lightweight and enhanced multiple feature extraction capabilities. The Spatial Receptive-field Convolution (SRConv) module was introduced as a convolutional module, effectively optimizing the performance of the convolutional neural network and improving overall model detection performance. The Region-Semantic Residual Module (RSR) module was proposed as an attention mechanism, which was fused with the C2f module to form the C2f_RSR module. This enhancement significantly enhanced the modeling ability to extract features pertinent to small defects. To reduce the sensitivity of the model to deviations in the position of small objects, The Gaussian Wasserstein Distance (GWD) metric was introduced into LGWD as a loss function. This adjustment has demonstrated outstanding performance in small target detection. Experimental results demonstrated that SR-NET enhanced mAP by 5.6% compared with baseline model. Simultaneously, SR-NET kept parameters and FLOPs almost unchanged. SR-NET proved to be well suited for practical production applications, and is able to meet real-time detection requirements.
Get full access to this article
View all access options for this article.
