摘要

针对印刷电路板(PCB, Printed Circuit Board)缺陷种类多、尺寸小、难以检测的问题,在YOLOv5的基础上提出一种新的PCB缺陷检测方法,重新设计加强特征网路,加强特征信息融合;引入注意力机制,使模型可以选择性过滤冗余信息、充分利用重要特征,提高模型对小尺寸缺陷的检测能力;引入Transformer结构,增强模型捕捉全局信息的能力,提高模型的准确率,最后得到新的模型记为CT-YOLOv5。经试验发现,改进后的模型准确率可达95.51%,提高了模型的准确率,可满足PCB缺陷检测要求。