HMA-DETR: a real-time hybrid multi-scale adaptive feature learning network for steel surface defect detection
摘要
To address the challenges of large-scale variations, weak textures, and parameter redundancy in steel surface defect detection, a real-time lightweight hybrid multi-scale adaptive feature learning network, termed HMA-DETR, is proposed based on RT-DETR. First, a hybrid gated aggregation network (HGANet) is designed to enhance defect-related spatial–channel representation through multi-scale spatial modeling and lightweight channel refinement, thereby improving the representation of defects with diverse shapes and textures. Second, a triple re-parameterized reconstruction stack (TriRepStack) is introduced after multi-scale feature fusion to reconstruct fused features and reduce semantic discrepancies across different feature levels, while preserving inference efficiency through structural re-parameterization. Finally, an adaptive sampling convolution (ASConv) is introduced to perform learnable offset-guided sampling and feature rearrangement during spatial reduction, aiming to alleviate information loss for small and irregular defects. Experimental results on the GC10-DET and NEU-DET datasets show that HMA-DETR improves mAP