基于YOLO-PCB的印刷电路板裸板缺陷检测

分享到

【获取途径】超星期刊网
【作者】王龙业，黄鋆，曾晓莉
【作者单位】¹西南石油大学电气信息学院；²西藏大学信息科学技术学院
【年份】2024
【卷号】第24卷
【期号】第15期
【页码】6338-6345
【ISSN】1671-1815
【关键词】印刷电路板 YOLOv5 小目标检测注意力机制
【摘要】针对当前印刷电路板裸板缺陷检测算法对小目标检测准确率较低、误检率过高等问题，提出一种改进的YOLO-PCB缺陷检测算法。该算法在YOLOv5s算法的基础上引入注意力机制，增强特征图的通道特征；同时引入加权双向特征金字塔网络改进特征融合层，使网络实现更高层次的特征融合；而且增加小目标检测层，提高网络对印刷电路板上小目标缺陷的检测能力。实验结果表明，相较于原YOLOv5算法，改进后的检测算法具有更强的特征提取融合能力和更高的检测精度，YOLO-PCB算法的mAP0.5提升了4.08%,mAP0.5:...
【基金】国家自然科学基金
【文献类型】期刊