EDFW-YOLO Enhancing YOLOv8 for Quantitative Analysis of Surface Defects in Hot-Rolled Strips
DOI:
Author:
Affiliation:

School of Physics and Electronic Engineering

Clc Number:

Fund Project:

National Natural Science Foundation of China(No. 61961037)

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Detecting surface defects on steel, especially in complex loading environments, poses significant challenges. In response, we introduce EDFW-YOLO, an algorithm built upon YOLOv8 specifically designed for detecting surface defects on hot-rolled steel strips. Our method enhances multi-scale feature fusion through the incorporation of the multi-scale con-version module C2f-EMSC. Additionally, we elevate detection accuracy by integrating the Dynamic Head target detection head, the Focal Modulation module, and the WIoU_Loss bounding box regression function. Experimental results on the NEU-DET dataset demonstrate that our optimized YOLOv8 model achieves an average accuracy (mAP) of 77.7%, with a 5.2% increase in network constraint rate. To adapt to different operating environments, it further improved the average accuracy (mAP) to 78.5% through data enhancement. Verification results on PCB defect data show that the algorithm has excellent generalization ability. This optimized algorithm significantly improves the extraction and fusion of surface defect features on hot-rolled strip steel and serves as a valuable reference for surface defect detection in alloy materials.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 29,2024
  • Revised:October 05,2024
  • Adopted:October 23,2024
  • Online:
  • Published: