基于树莓派和YOLOv5的PCB瑕疵检测

针对PCB瑕疵检测问题，提出了基于YOLOv5s的轻量化PCB瑕疵检测算法，并基于树莓派平台搭建了一套PCB瑕疵自动检测系统。首先，在Backbone阶段使用改进的空间金字塔池化代替原有的C3网络；其次，在Backbone与Neck中引入残差结构，并在小目标检测层面加入CBAM注意力机制；最后，将所提轻量化算法部署到树莓派上，并使用NCS2套件进行辅助加速，通过摄像头进行自动检测。通过测试，所提算法检测PCB瑕疵mAP达到99.1%,与原YOLOv5s模型相比，Params为其23%,FLOPs为其21%,PCB瑕疵检测系统运行速度达到7 fps,满足自动检测要求。