Improved PCB Chip Recognition Method Based on YOLOv7-tiny
To address the inefficiency of manual visual inspection in PCB chip detection, this paper proposes an improved PCB chip recognition method based on YOLOv7-tiny, leveraging a machine vision platform and deep learning techniques. While YOLOv7-tiny is effective for PCB detection, it suffers from high miss rates, poor performance in small object detection, and deployment challenges. To overcome these issues, we introduce the lightweight convolutional neural network RepViT as the backbone, enhancing the network’s ability to capture fine-grained features while reducing deployment costs on resource-constrained mobile devices. Additionally, we replace the SE attention mechanism in RepViT with the more lightweight and efficient ECA module to reduce computational complexity, and incorporate DSConv to further decrease the computational load, thereby improving inference speed. Meanwhile, using a dataset of 3076 images from both publicly available and laboratory sources, the proposed method achieved an average precision of 97.2% for single-class PCB chip detection, representing a 0.8% improvement over YOLOv7-tiny. Moreover, despite a 15.9% increase in FLOPs, the model’s parameter count was reduced by approximately 45%, with significant enhancements in generalization performance compared to the original network.
Authors
- Yuan Meng (ORCID: https://orcid.org/0000-0001-6468-8623)
- Zhengwei Yue (ORCID: https://orcid.org/0009-0003-5181-6034)
- Yushun Fan (ORCID: https://orcid.org/0000-0002-0071-4893)
- Zeyang Ma (ORCID: https://orcid.org/0000-0002-0390-1547)
- Xun Li (ORCID: https://orcid.org/0000-0002-7970-0370)
Institutions
- Xi'an Polytechnic University (CN)
Publication Details
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1109/icmee63700.2024.11025352
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00