Liu Jiaxin, Kang Bingyu, Liu Chao, Peng Xunhui, Bai Yan
College of Computer and Control Engineering, Northeast Forestry University, Harbin 150040, China.
Sensors (Basel). 2024 Sep 19;24(18):6055. doi: 10.3390/s24186055.
The small area of a printed circuit board (PCB) results in densely distributed defects, leading to a lower detection accuracy, which subsequently impacts the safety and stability of the circuit board. This paper proposes a new YOLO-BFRV network model based on the improved YOLOv8 framework to identify PCB defects more efficiently and accurately. First, a bidirectional feature pyramid network (BIFPN) is introduced to expand the receptive field of each feature level and enrich the semantic information to improve the feature extraction capability. Second, the YOLOv8 backbone network is refined into a lightweight FasterNet network, reducing the computational load while improving the detection accuracy of minor defects. Subsequently, the high-speed re-parameterized detection head (RepHead) reduces inference complexity and boosts the detection speed without compromising accuracy. Finally, the VarifocalLoss is employed to enhance the detection accuracy for densely distributed PCB defects. The experimental results demonstrate that the improved model increases the mAP by 4.12% compared to the benchmark YOLOv8s model, boosts the detection speed by 45.89%, and reduces the GFLOPs by 82.53%, further confirming the superiority of the algorithm presented in this paper.
印刷电路板(PCB)的面积较小,导致缺陷分布密集,检测精度较低,进而影响电路板的安全性和稳定性。本文提出了一种基于改进的YOLOv8框架的新型YOLO-BFRV网络模型,以更高效、准确地识别PCB缺陷。首先,引入双向特征金字塔网络(BIFPN)来扩大每个特征层的感受野并丰富语义信息,以提高特征提取能力。其次,将YOLOv8主干网络提炼为轻量级的FasterNet网络,在提高微小缺陷检测精度的同时降低计算量。随后,高速重参数化检测头(RepHead)降低推理复杂度并提高检测速度,且不影响准确性。最后,采用变焦距损失(VarifocalLoss)来提高密集分布的PCB缺陷的检测精度。实验结果表明,改进后的模型与基准YOLOv8s模型相比,mAP提高了4.12%,检测速度提高了45.89%且GFLOPs降低了82.53%,进一步证实了本文所提算法的优越性。