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