Fabric defect detection based on anchor-free network

Abstract

Fabrics play a pivotal role in human life and production, and surface defects can directly affect the quality and value of fabrics. Many methods for fabric defect detection have been proposed, but tiny defects are still difficult to be detected effectively, and the accuracy of defect localization and classification is low. To address these issues, a modified YOLOX network called YOLOX-CATD is proposed, which was supplemented with a coordinate attention module (CAM) and tiny defect detection layer (TDDL) for fast and efficient detection of fabric defects, especially tiny defects. Firstly, the anchor-free network is used as the detection framework to avoid the influence of hyperparameters of the setting anchor. Secondly, a CAM is proposed to enhance the representation of the object of interest in the input feature map and suppress the background regions. Finally, a TDDL is added to introduce high-resolution features to improve the localization accuracy of tiny defects. The experimental results on the Aliyun Tianchi Fabric dataset and NEU-DET demonstrate the superiority and generalization of the modified model. The mean average precision (mAP) of YOLOX-CATD on the fabric defect dataset is improved by 5.67% compared to the original YOLOX, and the detection speed can reach 35–36 frames per second (FPS). This proves that YOLOX-CATD can obtain excellent fabric defect detection performance and meet the urgent need for real-time detection in industrial applications.