journal article·2024-09-07Open access

Fabric Defect Detection Based on Improved Lightweight YOLOv8n

Shuangbao Ma, Yuna Liu, Yapeng Zhang

Applied Sciences· Volume 14· Issue 17

Abstract

In response to the challenges posed by complex background textures and limited hardware resources in fabric defect detection, this study proposes a lightweight fabric defect detection algorithm based on an improved GSL-YOLOv8n model. Firstly, to reduce the parameter count and complexity of the YOLOv8n network, the GhostNet concept is used to construct the C2fGhost module, replacing the conventional convolution layers in the YOLOv8n structure with Ghost convolutions. Secondly, the SimAM parameter-free attention mechanism is embedded at the end of the backbone network to eliminate redundant background, enhance semantic information for small targets, and improve the network’s feature extraction capability. Lastly, a lightweight shared convolution detection head is designed, employing the scale layer to adjust features, ensuring the lightweight nature of the model while minimizing precision loss. Compared to the original YOLOv8n model, the improved GSL-YOLOv8n algorithm increases the mAP@0.5 by 0.60% to 98.29% and reduces model size, computational load, and parameter count by 66.7%, 58.0%, and 67.4%, respectively, meeting the application requirements for fabric defect detection in textile industry production.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Industrial Vision Systems and Defect Detection · Image Processing Techniques and Applications · Advanced Measurement and Detection Methods

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Hubei Key Laboratory of Digital Textile Equipment, Wuhan Textile University, Wuhan 430073, China

School of Mechanical Engineering and Automation, Wuhan Textile University, Wuhan 430073, China

Funding

Fundamental Research Program of the National Natural Science Foundation of China

References (20)

Show source-provided references
1. Surface defect detection method of carbon fiber prepreg based on machine vision
2. An effective automatic fabric defect detection system using digital image processing10.13074/jent.2017.03.171241 ↗
3. Faster R-CNN: Towards real-time object detection with region proposal networks
4. Kim, J.H., Kim, N., and Won, C.S. (2023, January 4–10). High-speed drone detection based on YOLOv8. Proceedings of the IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Rhodes Island, Greece.10.1109/ICASSP49357.2023.10095516 ↗
5. Qi, J., Nguyen, M., and Yan, W. (2022, January 12–14). Waste classifcation from digital images using ConvNeXt. Proceedings of the Pacifc-Rim Symposium on Image and Video Technology, Virtual Event.10.1007/978-3-031-26431-3_1 ↗
6. Fabric defect detection algorithm based on improved Faster R-CNN
7. An Improved Fabric Defect Detection Method Based on SSD10.14504/ajr.8.S1.22 ↗
8. Guo, Y.B., Kang, X.J., Li, J.F., and Yang, Y.X. (2023). Automatic Fabric Defect Detection Method Using AC-YOLOv5. Electronics, 12.10.3390/electronics12132950 ↗
9. Fan, H., Zhu, D., and Li, Y. (2021, January 5–7). An improved yolov5 marine biological object detection algorithm. Proceedings of the 2021 2nd International Conference on Artificial Intelligence and Computer Engineering (ICAICE), Hangzhou, China.10.1109/ICAICE54393.2021.00014 ↗
10. Fabric defect detection using the improved YOLOv3 model
11. Research on fabric defect detection method based on lightweight network
12. Liu, B.B., Wang, H.Y., Cao, Z.F., Wang, Y., Tao, L., Yang, J., and Zhang, K. (2024). PRC-Light YOLO: An Efficient Lightweight Model for Fabric Defect Detection. Appl. Sci., 14.10.3390/app14020938 ↗
13. Saffron picking recognition in complex environments based on improved YOLOv7
14. Generalized focal loss: Learning qualified and distributed bounding boxes for dense object detection
15. Detection of Oscillations in Process Control Loops from Visual Image Space Using Deep Convolutional Networks10.1109/JAS.2023.124170 ↗
16. ICAPD framework and simAM-YOLOv8n for student cognitive engagement detection in classroom10.1109/ACCESS.2023.3337435 ↗
17. Fu, H., Song, G., and Wang, Y. (2021). Improved YOLOv4 marine target detection combined with CBAM. Symmetry, 13.10.3390/sym13040623 ↗
18. Coordinate-Aware Mask R-CNN with Group Normalization: A underwater marine animal instance segmentation framework10.1016/j.neucom.2024.127488 ↗
19. ES-Net: Efficient Scale-Aware Network for Tiny Defect Detection
20. A proxy learning curve for the Bayes classifier10.1016/j.patcog.2022.109240 ↗