基于卷积神经网络的PCB板通用分拣算法

PCB Board Common Sorting Algorithm Based on Convolutional Neural Network

传统PCB板分拣算法特征点由人工提取,位姿计算方法单一,且存在预处理步骤繁琐、样本要求高、位姿精度低、适用性低等缺点。为解决上述问题,文中提出一种基于卷积神经网络的PCB板通用分拣算法。改进的CaffeNet网络模型自动学习大量图片数据中PCB板的深层特征,并完成PCB板自动识别分类。运用改良RANSACS算法提高ORB算法特征点匹配正确率,采用最小二乘法计算模板图片与待匹配图片间角度差,以实现各类PCB板快速定位。实验证明,算法分拣正确率达到99.35%,具有良好的准确率和分拣效率。

The feature points of the traditional PCB board sorting algorithm are manually extracted,and the pose calculation method is single,which has the disadvantages of complex pretreatment steps,high sample requirements,low pose accuracy and low applicability.To solve the above problems,a general sorting algorithm for PCB board based on convolution neural network was proposed in this paper.An improved CaffeNet network model was constructed to automatically learn the deep features of PCB boards in a large number of image data,completed the automatic recognition and classification of PCB boards.The modified RANSACS algorithm was used to improve the accuracy of feature point matching of ORB algorithm,and the least square method was conducted to calculate the angle difference between the image of the board and the image to be matched,so as to realize the rapid positioning of all kinds of PCB boards.The result showed that the sorting accuracy of the algorithm reached 99.35%,which had good accuracy and sorting efficiency.