An Optimization Method for PCB Surface Defect Detection Model Based on Measurement of Defect Characteristics and Backbone Network Feature Information

被引:0
作者
Liu, Huixiang [1 ]
Zhao, Xin [1 ]
Liu, Qiong [1 ,2 ]
Chen, Wenbai [1 ]
机构
[1] Beijing Informat Sci & Technol Univ, Sch Automat, Beijing 100192, Peoples R China
[2] Beihang Univ, Jiangxi Res Inst, Nanchang 330096, Peoples R China
基金
中国国家自然科学基金;
关键词
defect detection; PCB; explainability; YOLOv8; target characteristics;
D O I
10.3390/s24227373
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
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.
引用
收藏
页数:20
相关论文
共 44 条
[1]   Speeded-Up Robust Features (SURF) [J].
Bay, Herbert ;
Ess, Andreas ;
Tuytelaars, Tinne ;
Van Gool, Luc .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2008, 110 (03) :346-359
[2]  
Baygin M, 2017, 2017 INTERNATIONAL ARTIFICIAL INTELLIGENCE AND DATA PROCESSING SYMPOSIUM (IDAP)
[3]   BRIEF: Binary Robust Independent Elementary Features [J].
Calonder, Michael ;
Lepetit, Vincent ;
Strecha, Christoph ;
Fua, Pascal .
COMPUTER VISION-ECCV 2010, PT IV, 2010, 6314 :778-792
[4]  
Carion Nicolas, 2020, Computer Vision - ECCV 2020. 16th European Conference. Proceedings. Lecture Notes in Computer Science (LNCS 12346), P213, DOI 10.1007/978-3-030-58452-8_13
[5]   TDD-net: a tiny defect detection network for printed circuit boards [J].
Ding, Runwei ;
Dai, Linhui ;
Li, Guangpeng ;
Liu, Hong .
CAAI TRANSACTIONS ON INTELLIGENCE TECHNOLOGY, 2019, 4 (02) :110-116
[6]   YOLO-MBBi: PCB Surface Defect Detection Method Based on Enhanced YOLOv5 [J].
Du, Bowei ;
Wan, Fang ;
Lei, Guangbo ;
Xu, Li ;
Xu, Chengzhi ;
Xiong, Ying .
ELECTRONICS, 2023, 12 (13)
[7]   Improving PCB defect detection using selective feature attention and pixel shuffle pyramid [J].
Fung, Ka Chun ;
Xue, Kai -Wen ;
Lai, Cheung-Ming ;
Lin, Kwan-Ho ;
Lam, Kin-Man .
RESULTS IN ENGINEERING, 2024, 21
[8]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[9]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916
[10]  
Jocher G., 2023, YOLO by ultralytics