Improved YOLOv5-based surface defect detection technology for aluminum ingot alloys
DOI:
CSTR:
Author:
Affiliation:

1.Institute of Metrology and Testing,Guiyang 550003, China; 2.College of Electrical Engineering, Guizhou University,Guiyang 550025, China

Clc Number:

TP391; TG146; TN805

Fund Project:

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

    Aiming at the problems of irregular morphology and suboptimal detection performance of surface defect on aluminum ingot alloys, an improved YOLOv5-based defect detection method is proposed. Firstly, The Res2Net feature extraction network block is employed to replace the CSPDarknet53 module of the baseline model, which can effectively detect the multi-scale defect. Secondly, the CBAM convolutional attention module is introduced into the backbone network of YOLOv5 to enhance the representational ability of defect features. Finally, the over-parameterized reparameterization convolutional blocks are used to substitute for the 3×3 convolutional blocks in the backbone and neck networks so as to reduce the model′s inference latency. Experimental results compared with the traditional target detection methods demonstrate the improved method achieves a mAP of 75.8% for defect detection, which is a significant improvement both in detection accuracy and inference speed, and can well satisfy the tasks and demands of practical industrial production.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: November 22,2024
  • Published:
Article QR Code