Laser Journal, Volume. 45, Issue 10, 47(2024)

Real-time fabric defect detection algorithm based on YOLOv5s

JI Xunsheng... QIAN Fu* and DONG Yue |Show fewer author(s)
Author Affiliations
  • School of Internet of Things Engineering, Jiangnan University, Wuxi Jiangsu 214122, China
  • show less

    During the industrialized production of fabrics, fabric defects are varied and contain a large number of small target defects and elongated defects with extreme aspect ratios, which makes fabric defect detection a challenging task. To address this problem, an improved YOLOv5s algorithm is proposed in this paper. Firstly, the Mosaic data enhancement method is improved, which enriches the dataset while weakening the side effects of the original data enhancement method on the detection of some fabric defect types, and improves the detection of small targets and extreme aspect ratio defects. Then, the batch normalization is improved to representative batch normalization, which improves the algorithm’s differential feature representation of diverse defect instances and suppresses noise interference; finally, the lightweight coordinate attention is introduced, which encodes the long-distance dependency and channel dependency of the features with accurate location information, and enhances the algorithm’s ability to locate defects. The experimental results show that the algorithm in this paper significantly improves the detection ability of small targets and extreme aspect ratio defects, making the average detection accuracy mAP reach 81.3, which is 4.1% higher than the original YOLOv5s, and the detection speed is 32.6 fps, which fully meets the real-time requirements, and the algorithm better balances the detection accuracy and detection speed.

    Tools

    Get Citation

    Copy Citation Text

    JI Xunsheng, QIAN Fu, DONG Yue. Real-time fabric defect detection algorithm based on YOLOv5s[J]. Laser Journal, 2024, 45(10): 47

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Mar. 12, 2023

    Accepted: Jan. 2, 2025

    Published Online: Jan. 2, 2025

    The Author Email: Fu QIAN (643696927@qq.com)

    DOI:10.14016/j.cnki.jgzz.2024.10.047

    Topics