WHE-YOLO: A Lightweight PCB Surface Defect Detection Model Based on YOLOv8n
Detecting defects in printed circuit boards (PCBs) is crucial for ensuring the quality and reliability of electronic products. However, existing methods often struggle with complex processing methods and excessive parameters, which can lead to suboptimal performance in real-world applications. This paper proposes a novel approach, WHE-YOLO, which builds upon YOLOv8n by introducing a more efficient feature extraction mechanism (SPPELAN) and a lightweight neck structure (HSFPN). These innovations result in a 29 % reduction in model parameters and a 13 % decrease in computational costs while achieving a 3.4% improvement in mean Average Precision (mAP50). This work provides a more efficient and reliable scheme for PCB surface defect detection, with potential applications in real-time quality control systems.
Authors
- Dafeng Long (ORCID: https://orcid.org/0000-0002-2107-1444)
- Juncheng Zou
- Hongxiang Lin (ORCID: https://orcid.org/0000-0002-9813-7325)
- Guangyu Yang (ORCID: https://orcid.org/0000-0002-3612-8618)
- Junjie Chen
Institutions
- Huizhou University (CN)
Publication Details
- Published
- 2024-11-01
- DOI
- https://doi.org/10.1109/icacr62205.2024.11053741
- Citations
- 2
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.70