基于SMT-YOLOv8的PCB缺陷检测研究

针对PCB缺陷检测中目标尺寸小、权重文件庞大难以部署的问题,提出一种改进的YOLOv8小目标缺陷检测方法。该方法将SE注意力机制融入C2f中,使网络能够根据通道域的信息给图像不同位置赋予不同的权重,获取更重要的特征信息;在SPPF中引入Basic RFB以增强网络感受野,提升网络的特征提取能力;新增小目标检测尺度,提升模型对微小缺陷的检测能力;舍弃大目标检测尺度,降低计算负荷并缩小权重文件。实验结果表明,在公开的PCB缺陷数据集,改进后的YOLOv8较原算法平均精度提升了2.6%、权重文件缩小了27.3%、FPS达到34.4 ms/帧。