计算机与现代化 ›› 2024, Vol. 0 ›› Issue (12): 45-52.doi: 10.3969/j.issn.1006-2475.2024.12.007
摘要: 针对目前传统网络模型对PCB缺陷检测不准确、检测速度慢、识别精度低等问题, 提出了一种基于YOLOv7改进的PCB缺陷检测方法。首先,该方法在原YOLOv7模型基础上采用FasterNet中的部分卷积PConv代替CatConv,从而减少内存访问和参数量,提高检测速度;其次,将双向特征金字塔网络(BiFPN)引入到YOLOv7模型的头部网络,实现PCB缺陷的多尺度特征融合,提升模型的检测精度,再次,将FasterNet模块与BiFPN融合,形成YOLOv7+FasterNet+BiFPN的PCB缺陷检测模型,增强模型对缺陷的特征表达能力;最后,将原有的CIoU改进为XIoU损失函数,不仅提高了模型的收敛速度和对小目标边界框的扰动抵抗能力,也准确地衡量了边界框预测的准确性和定位精度。实验结果表明:改进后的YOLOv7模型在测试集上的mAP@0.5达到了95.7%,召回率为98.0%,相比于原YOLOv7模型mAP@0.5值和召回率分别提高了7、2个百分点,检测时间仅为21.7 ms,同时,FLOPs值与原模型相比计算量也减小了6.5 G。本文方法在检测速度、查全率和准确率等方面均优于传统网络,为PCB缺陷检测提供了有效的解决方案。
中图分类号: