U-SMR: U-SwinT & multi-residual network for fabric defect detection

被引:7
作者
Qu, Hao [1 ]
Di, Lan [1 ]
Liang, Jiuzhen [2 ]
Liu, Hao [2 ]
机构
[1] Jiangnan Univ, Sch Artificial Intelligence & Comp, Wuxi 214122, Peoples R China
[2] Changzhou Univ, Sch Comp Sci & Artificial Intelligence, Changzhou 213164, Peoples R China
关键词
Fabric defect detection; U-SMR net; Swin transformer; Dual-branch pyramid module; Recursive multi-residuals; SALIENT OBJECT DETECTION;
D O I
10.1016/j.engappai.2023.107094
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Fabric defect detection methods based on deep networks are widely used in the textile industry, but they often suffer from poor model generalization and blurry edge detection. To resolve these challenges, we propose a novel network called "U-SMR Net", which integrates global contextual features, defect detail features, and high-level semantic features through the combination of ResNet-50 and Swin Transformer modules. Our USMR network includes a lightweight multiscale feature extraction module, the dual-branch pyramid Module (DBPM), which is nested to preserve high-resolution, shallow semantic information. We propose a recursive multi-level residual decoding block for multiscale fusion to refine, filter, and enhance input characteristics, generating prediction maps at multiple stages, and by employing an improved binary cross entropy loss function to supervise saliency mapping. The experimental results based on four groups from ZJU-Leaper dataset demonstrate the superior performance of our approach compared to other competitive methods by achieving an average fmeav ������e score of 75.33%, and finally testing results from both ZJU-Leaper-Total dataset and the HKU-Fabric dataset further support our U-SMR Net's validity and generalization ability.
引用
收藏
页数:14
相关论文
共 51 条
[1]  
Achanta R, 2009, PROC CVPR IEEE, P1597, DOI 10.1109/CVPRW.2009.5206596
[2]  
Chen LC, 2017, Arxiv, DOI arXiv:1706.05587
[3]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[4]   Improved faster R-CNN for fabric defect detection based on Gabor filter with Genetic Algorithm optimization [J].
Chen, Mengqi ;
Yu, Lingjie ;
Zhi, Chao ;
Sun, Runjun ;
Zhu, Shuangwu ;
Gao, Zhongyuan ;
Ke, Zhenxia ;
Zhu, Mengqiu ;
Zhang, Yuming .
COMPUTERS IN INDUSTRY, 2022, 134
[5]   Contour-Aware Loss: Boundary-Aware Learning for Salient Object Segmentation [J].
Chen, Zixuan ;
Zhou, Huajun ;
Lai, Jianhuang ;
Yang, Lingxiao ;
Xie, Xiaohua .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :431-443
[6]  
Chen ZY, 2020, AAAI CONF ARTIF INTE, V34, P10599
[7]  
Chenkai Zhang, 2020, IEEE Transactions on Artificial Intelligence, V1, P219, DOI 10.1109/TAI.2021.3057027
[8]   A tutorial on the cross-entropy method [J].
De Boer, PT ;
Kroese, DP ;
Mannor, S ;
Rubinstein, RY .
ANNALS OF OPERATIONS RESEARCH, 2005, 134 (01) :19-67
[9]  
Deng ZJ, 2018, PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P684
[10]   Fabric Defect Detection Adopting Combined GLCM, Gabor Wavelet Features and Random Decision Forest [J].
Deotale, Nilesh Tejram ;
Sarode, Tanuja K. .
3D RESEARCH, 2019, 10 (01)