Surface defects in steel impact durability and availability, challenging industrial production. The similarity between small defects and the background hinders current detection methods from handling multi-scale variations and accurately locating defects, complicating recognition. To address these issues, we propose DMFSO-YOLO, a novel steel defect detection method. Within its backbone, the Speed-Optimized Precision Module (SOPM) enhances computational efficiency, reduces memory consumption, and mitigates overfitting. Additionally, the Dynamic Multi-scale Feature Fusion module (DMFF), designed in the YOLOv8 neck, improves feature extraction across dimensions and layers. The Normalized Gauss-Wasserstein Distance (NWD) loss function provides stable gradient feedback by accurately measuring the difference between predicted and actual bounding boxes. Experiments on the NEU-DET dataset show that DMFSO-YOLO achieves a 7.9% mAP improvement over traditional methods and reaches 245.2 FPS, highlighting its potential as a robust and efficient solution for real-time defect detection in industrial applications.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Dynamic Multi-Scale Fusion and Speed-Optimized Network for Steel Surface Defect Detection

  • Kelei Sun,
  • Mengwei Sun,
  • Huaping Zhou,
  • Bingwen Hu

摘要

Surface defects in steel impact durability and availability, challenging industrial production. The similarity between small defects and the background hinders current detection methods from handling multi-scale variations and accurately locating defects, complicating recognition. To address these issues, we propose DMFSO-YOLO, a novel steel defect detection method. Within its backbone, the Speed-Optimized Precision Module (SOPM) enhances computational efficiency, reduces memory consumption, and mitigates overfitting. Additionally, the Dynamic Multi-scale Feature Fusion module (DMFF), designed in the YOLOv8 neck, improves feature extraction across dimensions and layers. The Normalized Gauss-Wasserstein Distance (NWD) loss function provides stable gradient feedback by accurately measuring the difference between predicted and actual bounding boxes. Experiments on the NEU-DET dataset show that DMFSO-YOLO achieves a 7.9% mAP improvement over traditional methods and reaches 245.2 FPS, highlighting its potential as a robust and efficient solution for real-time defect detection in industrial applications.