YOLOv5-GXT: A New PCB Surface Defect Detection Algorithm Based on YOLOv5.

Zhicong Zhang,Fanglin Chen,Aoran Zhang,Gaoming Lin,Xiao Wang,Changjun Zhou
DOI: https://doi.org/10.1109/icaci58115.2023.10146129
2023-01-01
Abstract:Printed Circuit Board (PCB) surface defect detection is the most important step in the PCB manufacturing process. How to accurately and efficiently implement PCB quality inspection is still a challenging task in the field of automatic detection. This paper proposes an improved PCB surface defect detection algorithm YOLOv5-GXT based on YOLOv5. The main structure of the network includes GHostv2 Block in the backbone network to achieve feature map reduction and long-distance information capture, reduce redundant computing, and more efficient feature extraction; C3XTR was used to replace C3 before the output of the backbone network, to explore the global information induction with self-attention, to speed up the training of the model, and Mish has used to improve the performance of the model. A large number of experiments on the PKU-Market-PCB dataset show that the YOLOv5-GXT model achieves 98.4% map (Mean Average Precision) and 97% F1-Score, and has faster training speed and better detection performance than the YOLOv5 model.
What problem does this paper attempt to address?