Printed circuit board defect detection based on improved YOLOv5
DOI:
CSTR:
Author:
Affiliation:

1.School of Electrical Engineering and Automation, Tianjin University of Technology,Tianjin 300380, China; 2.Tianjin Hongci Subao Technology Co., Ltd.,Tianjin 300400, China

Clc Number:

TP391.9

Fund Project:

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

    To address the challenges of confusion among defect categories and the difficulty in detecting small defect targets in printed circuit board defect detection, an improved YOLOv5 detection model was proposed. The Swin-Transformer is incorporated into the backbone network to extract multi-scale features, capturing both local and global information. A prediction feature layer is added specifically for small targets, and the new multi-scale feature fusion and detection structure enable the model to learn more comprehensive feature information. The ECIoU_Loss is employed as the loss function, facilitating collaborative optimization between detection speed and accuracy in circuit board defect detection. Experimental results demonstrate that the improved YOLOv5 model achieves a mean average precision of 98.7% on the PCB Defect dataset, with a precision of 99.7% and a recall of 97.4%. These performance metrics outperform current mainstream detection models, showcasing the improved YOLOv5 model's effectiveness in classifying and localizing circuit board defects.

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