Abstract:International Journal of Pattern Recognition and Artificial Intelligence, Ahead of Print. Printed circuit boards (PCBs) play a critical role in electronic products. Ensuring these products' long-term reliability and consistent performance requires effective PCB defect detection. Although existing deep learning models for PCB defect detection are not highly accurate, they often neglect capability considerations. This paper introduces a precise, fast, and lightweight defect detection model, CCG-YOLO, based on an enhanced YOLOv5 model to address this issue. The enhancements in CCG-YOLO can be summarized as follows: (1) Improved Backbone network: The feature extraction ability of the Backbone network is enhanced by introducing a C3HB module, which fosters spatial interaction capabilities. (2) Lightweight feature fusion network: A lightweight convolution structure called Ghost-Shuffle Convolution is incorporated in the feature fusion network, remarkably reducing model parameters while maintaining performance. (3) Efficient residual networking: To enhance model performance further, a CNeB module is introduced based on the ConvNeXt network, which replaces the C3 module in the Neck. CNeB improves model detection accuracy and reduces the number of model parameters. The combination of these enhancements results in impressive performance. CCG-YOLO achieves mean average precision (mAP@0.5) of 99.5% and 88.75% in mAP@0.5:0.95 on the TDD-Net public dataset. Compared with the original YOLOv5s algorithm, CCG-YOLO offers a 4.24% improvement in mAP@0.5:0.95, a 1[math]MB reduction in model size, a 0.472[math]M decrease in the number of parameters, a 0.6G floating point operation reduction in computational complexity, and a 120 frames per second real-time inference speed. These experimental results underscore that the proposed model excels in accuracy and speed and has a compact size for PCB defect detection. Moreover, CCG-YOLO is easily deployable on low-end devices, making it well-suited for meeting the real-time requirements of industrial defect detection.

Chip Defect Detection Based on Deep Learning Method

Integrated Circuit Packaging Defect Analysis and Deep Learning Detection Method

Small-scale defect detection in industrial environment based on lightweight deep learning network

Attention-based deep learning for chip-surface-defect detection

PCB Defect Detection Based on Improved Deep Learning Model

A Machine Learning Approach for Automated Detection of Critical PCB Flaws in Optical Sensing Systems

A Simple Confidence-Supervised Model for High-Resolution Defect Recognition

Automated Detection of Defects with Low Semantic Information in X-ray Images Based on Deep Learning

Defect Detection in Printed Circuit Boards Using You-Only-Look-Once Convolutional Neural Networks

PCB Electronic Component Defect Detection Method based on Improved YOLOv4 Algorithm

Small object detection method with shallow feature fusion network for chip surface defect detection

LW-YOLO: Lightweight Deep Learning Model for Fast and Precise Defect Detection in Printed Circuit Boards

PCB Defect Detection Based on Image Processing and Improved YOLOv5

Applying Deep Learning to Construct a Defect Detection System for Ceramic Substrates

Machine Vision Product Appearance Defect Detection Based on Deep Learning

YOLO-pdd: A Novel Multi-scale PCB Defect Detection Method Using Deep Representations with Sequential Images

PCB Defect Detection Using Deep Learning Methods

A Design of Bare Printed Circuit Board Defect Detection System Based on YOLOv8

A Particleboard Surface Defect Detection Method Research Based on the Deep Learning Algorithm

Efficient Micro-LED defect detection based on microscopic vision and deep learning