PCEB-YOLOV8: An Improved PCB Defect Detection Method Based on YOLOV8
Printed circuit board defect recognition is of great significance to ensuring the quality of electronic products. This paper proposes an improved PCB defect detection method based on YOLOV8. Firstly, the data augmentation method PredMix (prediction mix) is adopted to simulate scenarios such as overlapping and occlusion in images, where the displays are incomplete or blurred, thereby enhancing the robustness of the network model. Secondly, BiFPN is used to replace PAN-FPN in the fusion layer to strengthen the feature fusion capability. Finally, the CESE_C2f module is employed in the Backbone part, leveraging a lightweight attention mechanism to improve the detection effect while reducing the number of parameters. Experimental results show that compared with the original YOLOV8, the MAP value of this improved algorithm reaches $\mathbf{92.6}\%$, with a $\mathbf{2.3\%}$ increase in accuracy.
Authors
- Chunhe Song (ORCID: https://orcid.org/0000-0001-8392-1777)
- Xia Yu (ORCID: https://orcid.org/0000-0002-8401-0462)
- Meihui Li (ORCID: https://orcid.org/0000-0002-2727-1391)
- Rui Li (ORCID: https://orcid.org/0000-0003-2476-9909)
- Mengmeng Chu
Institutions
- Shenyang Institute of Automation (CN)
- Shenyang University of Technology (CN)
- Chinese Academy of Sciences (CN)
Publication Details
- Published
- 2024-10-25
- DOI
- https://doi.org/10.1109/icemce64157.2024.10862850
- Citations
- 3
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 1.05
Funders
- National Natural Science Foundation of China