Multi-scale sensing and multi-dimensional feature enhancement for surface defect detection of hot-rolled steel strip

Surface defect detection of hot-rolled steel strip is a crucial process in its production. Aiming at the problem of poor detection accuracy caused by complex shape of defect features and imbalance between object scale and feature scale, this paper proposes a multi-scale sensing and multi-dimensional feature enhancement algorithm. Firstly, a multi-branch residual aggregation module (MRAM) is proposed, which uses the interactive features of different receptive fields between layers to capture global and local texture features, improving the feature extraction capability for defects of different scales. Then, the multi-dimensional attention enhancement head (MAEH) is proposed, which integrates feature information from different prediction branches and realises feature information enhancement in scale, space and channel dimensions for the aggregated information to improve the accuracy and robustness of steel surface defect detection. The results show that the proposed algorithm achieves 82.7% mAP and 89 FPS on the NEU-DET dataset, and 73.3% mAP and 94 FPS on the GC10-DET dataset. Compared with state-of-the-art algorithms, the proposed algorithm has the highest mAP, strong generalisation ability, and good real-time performance, which proves its effectiveness in detecting hot-rolled steel strip surface defects.