학술논문

A Lightweight Detector Based on Attention Mechanism for Fabric Defect Detection
Document Type
article
Source
IEEE Access, Vol 11, Pp 33554-33569 (2023)
Subject
Fabric defect detection
SoftPool
attention mechanism
depthwise separable convolution
lightweight
Electrical engineering. Electronics. Nuclear engineering
TK1-9971
Language
English
ISSN
2169-3536
Abstract
Defects on fabric surfaces are difficult to identify owing to unsuitable computing devices, highly complex algorithms, small size, and high degree of integration with the fabric. To this end, this study proposes a lightweight fabric defect-detection network, YOLO-SCD, based on attention mechanism. The introduction of depth-wise separable convolution and the attention mechanism enhanced the capacity of the neck network to extract the defective features and increased the detection speed of the overall network. The extensive experimental results revealed that YOLO-SCD achieved an average accuracy of 82.92%, effective improvement of 8.49% in mAP, and an improvement of 37 fps compared to the original YOLOv4 on a standard fabric defect dataset. By leveraging its swift detection speed and high efficiency, YOLO-SCD excels in both the general fabric defect category and the difficult-to-detect fabric. Overall, it exhibited strong performance in detecting both minor flaws and flaws with high fabric integration. Furthermore, the proposed model was extended to steel datasets with similar characteristics.