AYOLOv7-tiny: Towards efficient defect detection in solid color circular weft fabric

被引:12
作者
Kang, Xinjian [1 ]
Li, Junfeng [1 ]
机构
[1] Zhejiang Sci Tech Univ, Hangzhou 310018, Peoples R China
关键词
Fabric defect detection; machine vision; YOLOv7-tiny; R-CNN; DEEP;
D O I
10.1177/00405175231205898
中图分类号
TB3 [工程材料学]; TS1 [纺织工业、染整工业];
学科分类号
0805 ; 080502 ; 0821 ;
摘要
This article presents a solid color circular weft fabric defect detection method based on AYOLOv7-tiny. The aim of the development of this network is to enable real-time defect detection in the production of large circular machine fabrics. The network has a good accuracy rate, fast detection speed, and a lightweight model. In the YOLOv7-tiny network, a space-to-depth layer followed by a non-strided convolution layer is introduced to enhance the feature extraction capability, improve image sharpness, address issues such as uneven grayscale and difficult detection of minor defects, and simplify the model complexity while reducing computation. Additionally, we combined the Squeeze and Excitation (SE) and Spatial Attention Module (SAM) based on Convolutional Block Attention Module (CBAM) to construct the Hybrid Attention Module (SC), which is integrated into the YOLOv7-tiny network. The SC module increases the weight of important features, enhances the feature extraction capability, improves image segmentation, and enhances the accuracy of the network's location information. Through extensive experiments with a dataset of large circular machine solid color circular weft fabric defect collected from an industrial site, the results show that AYOLOv7-tiny has a detection accuracy (mean average precision) of 98.7%, a detection speed of up to 333 frames per second, and a computational complexity of only 4.4 GFlops, which is better than the current mainstream surface defect detection models. The detection accuracy, detection speed, and model complexity of the AYOLOv7-tiny network all meet the real-time detection requirements of the industry and have been successfully used for real-time detection of large circular machine fabric defects.
引用
收藏
页码:225 / 245
页数:21
相关论文
共 36 条
[21]   Deep Learning in Sheet Metal Bending With a Novel Theory-Guided Deep Neural Network [J].
Liu, Shiming ;
Xia, Yifan ;
Shi, Zhusheng ;
Yu, Hui ;
Li, Zhiqiang ;
Lin, Jianguo .
IEEE-CAA JOURNAL OF AUTOMATICA SINICA, 2021, 8 (03) :565-581
[22]   Deep Learning-Accelerated Designs of Tunable Magneto-Mechanical Metamaterials [J].
Ma, Chunping ;
Chang, Yilong ;
Wu, Shuai ;
Zhao, Ruike Renee .
ACS APPLIED MATERIALS & INTERFACES, 2022, 14 (29) :33892-33902
[23]   Automatic recognition of woven fabric structural parameters: a review [J].
Meng, Shuo ;
Pan, Ruru ;
Gao, Weidong ;
Yan, Benchao ;
Peng, Yangyang .
ARTIFICIAL INTELLIGENCE REVIEW, 2022, 55 (08) :6345-6387
[24]   Deep Learning-based Text Classification: A Comprehensive Review [J].
Minaee, Shervin ;
Kalchbrenner, Nal ;
Cambria, Erik ;
Nikzad, Narjes ;
Chenaghlu, Meysam ;
Gao, Jianfeng .
ACM COMPUTING SURVEYS, 2022, 54 (03)
[25]   SSDMNV2: A real time DNN-based face mask detection system using single shot multibox detector and MobileNetV2 [J].
Nagrath, Preeti ;
Jain, Rachna ;
Madan, Agam ;
Arora, Rohan ;
Kataria, Piyush ;
Hemanth, Jude .
SUSTAINABLE CITIES AND SOCIETY, 2021, 66 (66)
[26]   Automatic Fabric Defect Detection Method Using PRAN-Net [J].
Peng, Peiran ;
Wang, Ying ;
Hao, Can ;
Zhu, Zhizhong ;
Liu, Tong ;
Zhou, Weihu .
APPLIED SCIENCES-BASEL, 2020, 10 (23) :1-13
[27]   Fabric Defect Detection Using Computer Vision Techniques: A Comprehensive Review [J].
Rasheed, Aqsa ;
Zafar, Bushra ;
Rasheed, Amina ;
Ali, Nouman ;
Sajid, Muhammad ;
Dar, Saadat Hanif ;
Habib, Usman ;
Shehryar, Tehmina ;
Mahmood, Muhammad Tariq .
MATHEMATICAL PROBLEMS IN ENGINEERING, 2020, 2020
[28]   No More Strided Convolutions or Pooling: A New CNN Building Block for Low-Resolution Images and Small Objects [J].
Sunkara, Raja ;
Luo, Tie .
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2022, PT III, 2023, 13715 :443-459
[29]   Improving the efficiency of single-cell genome sequencing based on overlapping pooling strategy and CNV analysis [J].
Tu, Jing ;
Yang, Zengyan ;
Lu, Na ;
Lu, Zuhong .
ROYAL SOCIETY OPEN SCIENCE, 2022, 9 (01)
[30]   YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors [J].
Wang, Chien-Yao ;
Bochkovskiy, Alexey ;
Liao, Hong-Yuan Mark .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, :7464-7475