Back to articles
ISAIMS 2023 Deep Learning
Volume: 68 | Article ID: 040402
Image
MSCA-YOLO: A YOLOv5-based Steel Defect Detection Method Enhanced with Multi-Scale Feature Extraction and Contextual Augmentation
  DOI :  10.2352/J.ImagingSci.Technol.2024.68.4.040402  Published OnlineJuly 2024
Abstract
Abstract

Steel surface defect detection in industrial quality control has always been a challenging objective detection task in the field of computer vision. However, unlike other detection problems, some surface defects on steel are relatively small compared to the entire inspection object, leading to less prominent defect features in the detection. To address these issues, we propose a YOLOv5-based steel defect detection method enhanced with multi-scale feature extraction and contextual augmentation (MSCA-YOLO). Specifically, adopting the YOLOv5 as the backbone network, we first add the C3-RFE to expand the receptive. Then, we design a neck network structure via combining multi-scale guided upsampling, which effectively enhances the model’s ability to handle multi-scale features and improves the model’s feature extraction ability for small defects. Finally, we propose a context mechanism that provides the model with a deeper context analysis capability, offering richer up-and-down information. The experiments on the NEU-DET dataset show that MSCA-YOLO achieves a mean Average Precision of 0.645 while maintaining rapid detection, especially at an Intersection over Union threshold of 0.5. It also exhibits substantial improvements in Precision compared to YOLOv5 across six defect types: Crazing (18.5% increase), Inclusion (1.2% increase), Patches (1.9% increase), Pitted_Surface (7.8% increase), Rolled-in_Scale (8.9% increase), and Scratches (6.5% increase). This achievement marks the efficiency and reliability of MSCA-YOLO in automated steel surface defect detection, providing a new solution for real-time inspection of steel surface defects.

Subject Areas :
Views 94
Downloads 7
 articleview.views 94
 articleview.downloads 7
  Cite this article 

Yao Wang, Chengxin Liang, Xiao Wang, Yushan Liu, "MSCA-YOLO: A YOLOv5-based Steel Defect Detection Method Enhanced with Multi-Scale Feature Extraction and Contextual Augmentationin Journal of Imaging Science and Technology,  2024,  pp 1 - 10,  https://doi.org/10.2352/J.ImagingSci.Technol.2024.68.4.040402

 Copy citation
  Copyright statement 
Copyright © Society for Imaging Science and Technology 2024
  Article timeline 
  • received December 2023
  • accepted February 2024
  • PublishedJuly 2024

Preprint submitted to:
  Login or subscribe to view the content