IM-DETR: DETR with Mix-Encoder for Industrial Scenarios

Industrial defect detection is a fundamental task in intelligent manufacturing, yet existing object detection methods often struggle with the characteristics of industrial defects, such as small size, irregular shapes, and complex visual backgrounds. Moreover, most detection models are designed primarily for natural image datasets, resulting in limited robustness when deployed in real-world industrial environments. To address these challenges, this research focuses on industrial defect detection and presents contributions at both the dataset and method levels. First, two real-world industrial defect datasets collected from actual production lines are introduced, namely, the Stator Housing Defect Dataset and the Cover Plate Silicone Defect Dataset, which cover representative inspection scenarios with distinct defect characteristics. Second, we propose a detection transformer with a mixed encoder for industrial scenarios (IM-DETR). By integrating heterogeneous multi-scale feature representations, the proposed framework jointly enhances local detail sensitivity and global contextual reasoning without relying on complex post-processing. Extensive experiments on the proposed industrial datasets demonstrate that IM-DETR consistently outperforms existing state-of-the-art detection methods, particularly in scenarios involving small defects, complex backgrounds, and appearance ambiguity, validating the effectiveness and robustness of the proposed approach.