MS-DETR: a real-time multi-scale detection transformer for PCB defect detection
摘要
Surface defects on printed circuit boards degrade product quality and impact device performance, stability, and reliability. Accurately identifying these defects amidst a complex background is challenging due to the compact PCB layout. To address this, we propose the Multi-Scale Detection Transformer (MS-DETR), designed to enhance detection accuracy while maintaining efficiency for resource-constrained environments. MS-DETR introduces a Multi-Stage Convolution module to replace traditional convolution layers, enhancing feature extraction for small objects while reducing model complexity. The Slim-Scale Adaptive Fusion architecture, incorporating Ghost Squeeze-and-Excitation Convolution and Scale Adaptive Fusion modules, reduces network computation and enhances semantic information extraction. Additionally, integrating high-frequency and low-frequency (Hilo) information improves fine-grained feature extraction and focuses on critical features, while Inner Generalized Intersection over Union optimizes the loss function to accelerate convergence and enhance detection accuracy. Experimental results on a PCB defect dataset and a self-constructed dataset show that MS-DETR achieves a mean Average Precision of 96.90%, reducing arithmetic operations by 11.7 GFLOPs and parameter count by 5.9 M compared to Real-time Detection Transformer, while maintaining a detection speed of 115 frames per second. These attributes make MS-DETR suitable for edge systems or embedded devices, balancing accuracy, computational cost, and speed effectively.