An Enhanced RT-DETR Model for Improved Fabric Defect Detection in Industrial Environments

Fabric defect detection is critical for quality control in the textile industry. However, current challenges include the scarcity of high-quality datasets, the difficulty of detecting small defects in complex environments, and suboptimal detection performance for elongated defects. To address these issues, we propose an improved Real-Time Detection Transformer (RT-DETR) model specifically designed for fabric defect detection in industrial settings. A custom fiber defect dataset was created with random augmentations for improved diversity. We designed the Global Attention-based Instance Feature Interaction (GA-AIFI) module to enhance local feature extraction, improving the model's detection capability for small defects. Additionally, to address the prevalence of elongated defects in industrial fabric production, we introduced the Aspect Ratio-aware Distance-IoU (AR-DIoU) loss function, which further improves bounding box localization accuracy. Experiments conducted on both the custom dataset and the MVTec dataset demonstrate significant improvements, with [email protected] increasing by 3.5% and 3.6%,respectively, confirming the model's effectiveness.