基于卷积神经网络的PCB缺陷检测

PCB defect detection based on convolutional neural network

印刷电路板(PCB)在生产制造中由于生产工序等问题易导致电路板存在瑕疵缺陷,为提高对电路板缺陷的检测效率,提出了一种基于卷积神经网络(CNN)的电路板缺陷检测网络。该检测网络基于YOLO v4网络进行优化改造,针对于PCB制作精密、复杂,各类缺陷不易检测的难点,在优化后的网络中加入了基于细粒度空间域的长距离全局注意力机制,同时在SPP模块的基础上进行特征图重组作为各YOLO检测头的输入。通过使用长距离注意力机制通道将浅层网络提取到的特征传递到深层网络中,并采用特征图重组的方式提升特征信息丰富度,从而提高对于PCB缺陷检测的精度。经实验分析,与各类经典CNN相比,在PCB缺陷检测任务中,该算法有较大优势,整体缺陷的平均检测精度均值(mAP)达到91.40%,适用于实际生产、检测环节。

In the production of printed circuit boards(PCB),the production process and other problems incur flaws and defects on the circuit board.In order to enhance the detection efficiency of circuit board defects,a circuit board defect detection network based on convolutional neural network(CNN)was proposed.The whole detection network was optimized and reconstructed based on the YOLO v4 network.Aiming at the difficulty of precise and complex PCB production and difficult detection of various defects,a long-distance global attention mechanism based on fine-grained spatial domain was added to the optimized network.At the same time,on the basis of the spatial pyramid pooling(SPP)module,the feature map was reorganized as the input of each YOLO detection head.The long-distance attention mechanism channel was adopted to transfer the features extracted from the shallow network to the deep network,and the feature map reorganization method was utilized to boost the richness of feature information,thereby improving the accuracy of PCB defect detection.After experimental analysis,compared with various classic convolutional neural networks,the proposed algorithm is greatly superior in PCB board defect detection tasks.The mean average precision(mAP)of the overall defects reaches 91.40%,which is suitable for actual production and testing links.