Laser & Optoelectronics Progress, Volume. 59, Issue 10, 1010005(2022)

Fabric Defect Classification Algorithm Based on Multi-Scale Feature Fusion of Spatial Attention

Zhiyong Song and Haipeng Pan*
Author Affiliations
  • School of Mechanical and Automatic Control, Zhejiang Sci-Tech University, Hangzhou 310018, Zhejiang , China
  • show less

    A fabric surface defect classification technique based on spatial attention multiscale feature fusion is designed to address the problem of low-classification accuracy caused by complex texture and varied defect kinds of the fabric surface. The multiscale pyramid pooling module is used to maintain the information integrity of the feature map, and the rich semantic information extracted from the high-level feature map is used as a priori information to guide the low-level features, realizing the fusion of high-level and low-level features; the improved spatial attention module is integrated into a convolutional neural network to enhance the differential expression of features. The improved class activation mapping method is used to obtain the defect classification information and location information. The fabric surface defect image is recognized and detected using data augmentation and transfer learning methods. The experimental results show that the proposed algorithm can effectively increase the accuracy of fabric defect classification and obtain defect location information without manual location labeling.

    Tools

    Get Citation

    Copy Citation Text

    Zhiyong Song, Haipeng Pan. Fabric Defect Classification Algorithm Based on Multi-Scale Feature Fusion of Spatial Attention[J]. Laser & Optoelectronics Progress, 2022, 59(10): 1010005

    Download Citation

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

    Category: Image Processing

    Received: Apr. 14, 2021

    Accepted: May. 18, 2021

    Published Online: May. 16, 2022

    The Author Email: Pan Haipeng (pan13989896598@163.com)

    DOI:10.3788/LOP202259.1010005

    Topics