IDEAS home Printed from https://ideas.repec.org/a/gam/jsusta/v15y2023i7p5963-d1111119.html
   My bibliography  Save this article

PCB-YOLO: An Improved Detection Algorithm of PCB Surface Defects Based on YOLOv5

Author

Listed:
  • Junlong Tang

    (School of Physics and Electronic Science, Changsha University of Science and Technology, Changsha 410114, China)

  • Shenbo Liu

    (School of Physics and Electronic Science, Changsha University of Science and Technology, Changsha 410114, China)

  • Dongxue Zhao

    (School of Physics and Electronic Science, Changsha University of Science and Technology, Changsha 410114, China)

  • Lijun Tang

    (School of Physics and Electronic Science, Changsha University of Science and Technology, Changsha 410114, China)

  • Wanghui Zou

    (School of Physics and Electronic Science, Changsha University of Science and Technology, Changsha 410114, China)

  • Bin Zheng

    (School of Computer and Communications Engineering, Changsha University of Science and Technology, Changsha 410114, China)

Abstract

To address the problems of low network accuracy, slow speed, and a large number of model parameters in printed circuit board (PCB) defect detection, an improved detection algorithm of PCB surface defects based on YOLOv5 is proposed, named PCB-YOLO, in this paper. Based on the K-means++ algorithm, more suitable anchors for the dataset are obtained, and a small target detection layer is added to make the PCB-YOLO pay attention to more small target information. Swin transformer is embedded into the backbone network, and a united attention mechanism is constructed to reduce the interference between the background and defects in the image, and the analysis ability of the network is improved. Model volume compression is achieved by introducing depth-wise separable convolution. The EIoU loss function is used to optimize the regression process of the prediction frame and detection frame, which enhances the localization ability of small targets. The experimental results show that PCB-YOLO achieves a satisfactory balance between performance and consumption, reaching 95.97% mAP at 92.5 FPS, which is more accurate and faster than many other algorithms for real-time and high-precision detection of product surface defects.

Suggested Citation

  • Junlong Tang & Shenbo Liu & Dongxue Zhao & Lijun Tang & Wanghui Zou & Bin Zheng, 2023. "PCB-YOLO: An Improved Detection Algorithm of PCB Surface Defects Based on YOLOv5," Sustainability, MDPI, vol. 15(7), pages 1-17, March.
    • Handle: RePEc:gam:jsusta:v:15:y:2023:i:7:p:5963-:d:1111119
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2071-1050/15/7/5963/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2071-1050/15/7/5963/
    Download Restriction: no
    ---><---

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Zhike Zhao & Songying Li & Caizhang Wu & Xiaobing Wei, 2023. "Research on the Rapid Recognition Method of Electric Bicycles in Elevators Based on Machine Vision," Sustainability, MDPI, vol. 15(18), pages 1-21, September.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jsusta:v:15:y:2023:i:7:p:5963-:d:1111119. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.