面向带钢表面缺陷检测的HDM-YOLO方法

    HDM-YOLO: A method for steel strip surface defect detection

    • 带钢作为重要工业材料,带钢表面质量直接影响产品性能与经济价值。针对表面缺陷检测中小目标识别困难、特征信息损失及模型复杂度高的问题,文中提出一种基于YOLOv13n的轻量高效检测模型HDM-YOLO。该方法首先通过分层互补特征混合模块H-RAMi (Hierarchical Reciprocal Attention Mixer)重构主干网络,实现多尺度特征高效融合;其次采用动态采样机制DySample (Dynamic Sample Module)自适应优化特征传递,减少信息损失;最后构建基于MBConv (Mobile Inverted Bottleneck Convolution)的轻量化检测头Detect_MBConv,平衡检测精度与速度。实验表明,在NEU-DET数据集上,HDM-YOLO取得81.2%的mAP@0.5,精度较基线提升2.7%,参数量与计算量分别降低37.7%和38.8%,推理速度达101.8 FPS。该模型在GC10-DET数据集上也表现出良好泛化能力,验证了其较高的工程应用价值。

       

      Abstract:
      Objective As the core pillar industry in China, strip steel is very important for automobile manufacturing, construction and other industries. Its surface quality directly affects the product reliability and market value. At present, surface defect detection faces three major challenges: first, the defect scale does not match the network receptive field, and small-size defects are easy to miss detection. Second, continuous down sampling leads to the loss of texture information, and low contrast defects are difficult to identify in complex background. Third, the traditional deep learning model has a large number of parameters and complex calculation, which is difficult to realize real-time processing in the resource constrained industrial line. These deficiencies lead to the decline of detection accuracy and efficiency, and the increase of cost. Therefore, there is an urgent need for a high-performance solution that can take into account the detection accuracy, reasoning speed and lightweight design.
      Methods The H-RAMi module is introduced into the backbone network to get through the detail and semantic feature channels by cross level feature calibration, accurately capture the micro defect information, and lay a high-quality feature foundation for subsequent detection; In view of the fact that downsampling is easy to lose key details such as defect edges and textures, which leads to missed detection, DySample dynamic upsampling component is used to reconstruct the feature space association through adaptive weights to complete the short board of previous features. Facing the dual requirements of high precision and low delay for the limited computing power at the industrial end, the detection head is designed with a lightweight structure of Detect_MBconv, which uses deep separable convolution and channel attention coordination to compress the computational overhead and parameters while maintaining the receptive field.
      Results and Discussions The experimental results on NEU-DET dataset show that the average accuracy of HDM-YOLO is 81.2%, which is superior to the mainstream lightweight detector in the same parameter range. The ablation experiment results are shown in Table 2, which verifies the independent contribution of each module: H-RAMi improves the mAP by 0.7% through cross-scale feature interaction, dysample improves the small target detection accuracy by 1.4% through fine detail retention, and detect_mbconv reduces the model complexity by 22.3%. Table 3 model comparison experiment results show that compared with the baseline model YOLOv13n, the accuracy of HDM-YOLO is improved by 2.7%, the number of parameters is reduced by 37.7%, the computational complexity is reduced by 38.8%, and the reasoning speed is improved from 63.6 frame/s to 101.8 frame/s, meeting the requirements of industrial real-time detection. The comparison of visualization results is shown in Figure 9, which intuitively shows the advantages of HDM-YOLO in small defect detection. The thermodynamic diagram comparison in Figure 10 further verifies the model's ability to accurately locate the defect area. The synergy of H-RAMi and DySample significantly improves the performance of small defect detection. Through the synergy of deep separable convolution and channel attention, it reduces a lot of computational overhead and parameters while maintaining the receptive field. In addition, table 5 GC10-DET comparative experimental results show that the HDM-YOLO mAP value has increased by 3.2%, and the detection speed has increased by 46.9 frames, showing good generalization ability and engineering application value.
      Conclusions HDM-YOLO provides an efficient and lightweight solution for strip surface defect detection, and effectively solves the core challenges of existing methods, such as low detection accuracy of small targets, loss of feature information and lack of real-time. By integrating hierarchical complementary feature mixing module (H-RAMi), dynamic sampling module (DySample) and lightweight detection head (Detect_MBConv), the model achieves an excellent balance between detection accuracy (NEU-DET dataset 81.2% mAP), reasoning speed (101.8 frame/s) and model compactness (1.5 M parameters). Visualization and thermal diagram analysis have confirmed the model's ability to accurately identify and locate the defect area. Its strong generalization ability and modular design on GC10-DET dataset enable it to flexibly adapt to various industrial detection scenarios, including high-speed strip production lines. In addition to steel manufacturing, the lightweight architecture of HDM-YOLO provides a useful reference for the development of computer vision models in other resource constrained industrial scenes. In general, the model has significant industrial deployment potential, which can improve the efficiency of quality control, reduce production costs, and promote the development of lightweight defect detection technology.

       

    /

    返回文章
    返回