Translation of steel surface defect detection algorithm with fusion of multiple attention detection heads
CSTR:
Author:
Affiliation:

1. School of Electrical Engineering and Automation, Tianjin University of Technology, Tianjin 300384, China;2. School of Electrical Engineering, North China University of Science and Technology, Tangshan 063210,China

Clc Number:

Fund Project:

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

    This paper proposes YOLOv5-CJ for steel surface defect detection. C3_MSBlock enhances multi-scale feature extraction and enlarges the receptive field, while DyHead introduces scale-, spatial-, and task-aware attention to improve robustness in complex scenes. Soft non-maximum suppression (NMS) further improves recognition in overlapping regions. Compared with YOLOv5s, YOLOv5-CJ improves the mean average precision at intersection over union (IoU) of 0.5 (mAP0.5) and the mean average precision averaged over IoU threshold from 0.5 to 0.95 (mAP0.5:0.95) by 1.9% and 7.2% on the Northeastern University steel surface defect (NEU-DET) dataset, and by 5.3% and 4.3% on the GC10 steel surface defect (GC10-DET) dataset, respectively, demonstrating its effectiveness for industrial defect detection.

    Reference
    Related
    Cited by
Get Citation

Tao SHI, Jie CUI, Song LI. Translation of steel surface defect detection algorithm with fusion of multiple attention detection heads[J]. Optoelectronics Letters,2026,22(8):475-480

Copy
Related Videos

Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:November 05,2023
  • Revised:July 02,2026
  • Adopted:
  • Online: August 24,2026
  • Published:
Article QR Code