An Optimization Method for PCB Surface Defect Detection Model Based on Measurement of Defect Characteristics and Backbone Network Feature Information
Huixiang Liu,Xin Zhao,Qiong Liu,Wenbai Chen
DOI: https://doi.org/10.3390/s24227373
IF: 3.9
2024-11-27
Sensors
Abstract:Printed Circuit Boards (PCBs) are essential components in electronic devices, making defect detection crucial. PCB surface defects are diverse, complex, low in feature resolution, and often resemble the background, leading to detection challenges. This paper proposes the YOLOv8_DSM algorithm for PCB surface defect detection, optimized based on the three major characteristics of defect targets and feature map visualization. First, to address the complexity and variety of defect shapes, we introduce CSPLayer_2DCNv3, which incorporates deformable convolution into the backbone network. This enhances adaptive defect feature extraction, effectively capturing diverse defect characteristics. Second, to handle low feature resolution and background resemblance, we design a Shallow-layer Low-semantic Feature Fusion Module (SLFFM). By visualizing the last four downsampling convolution layers of the YOLOv8 backbone, we incorporate feature information from the second downsampling layer into SLFFM. We apply feature map separation-based SPDConv for downsampling, providing PAN-FPN with rich, fine-grained shallow-layer features. Additionally, SLFFM employs the bi-level routing attention (BRA) mechanism as a feature aggregation module, mitigating defect-background similarity issues. Lastly, MPDIoU is used as the bounding box loss regression function, improving training efficiency by enhancing convergence speed and accuracy. Experimental results show that YOLOv8_DSM achieves a mAP (0.5:0.9) of 63.4%, representing a 5.14% improvement over the original model. The model's Frames Per Second (FPS) reaches 144.6. To meet practical engineering requirements, the designed PCB defect detection model is deployed in a PCB quality inspection system on a PC platform.
engineering, electrical & electronic,chemistry, analytical,instruments & instrumentation