Paper
16 August 2023 PCB defect detection algorithm based on improved YOLOv5
Rui Wu, Haibin Li
Author Affiliations +
Proceedings Volume 12787, Sixth International Conference on Advanced Electronic Materials, Computers, and Software Engineering (AEMCSE 2023); 127871J (2023) https://doi.org/10.1117/12.3004442
Event: 6th International Conference on Advanced Electronic Materials, Computers and Software Engineering (AEMCSE 2023), 2023, Shenyang, China
Abstract
Aiming at the problem of small contrast difference between the defective region and the background region in PCB images, This Article presents an enhanced YOLOv5 algorithm with a multi-scale detection approach. To enhance the YOLOv5 algorithm, this Article upgrades its backbone network by replacing Characteristics Abstraction network with the ASPP module. This change aims to improve the network's perceptual field and Characteristics extraction capability. Secondly, in order to improve the attention of the model to other regions, this Article introduces the attention mechanism Coordinate Attention module, which embeds the location information into the channel attention and achieves multi-scale processing and Characteristics fusion. Finally, this Article uses different sizes of anchor frames for multiscale detection of defective targets. The experimental results show that the size of the improved multiscale network model is only 83% of the size of the original YOLOv5 model, and the mAP on the dataset reaches 97.2%. The algorithm proposed in this Article can effectively detect various defects in PCB images and has high detection precision and low false detection rate, which has good practical value and prospect of popularization and application.
(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Rui Wu and Haibin Li "PCB defect detection algorithm based on improved YOLOv5", Proc. SPIE 12787, Sixth International Conference on Advanced Electronic Materials, Computers, and Software Engineering (AEMCSE 2023), 127871J (16 August 2023); https://doi.org/10.1117/12.3004442
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Defect detection

Detection and tracking algorithms

Image segmentation

Target detection

Convolution

Back to Top