一种基于注意力机制的YOLO缺陷检测算法

Defect Inspection Algorithm Using YOLO Based on Attention Mechanism

针对零部件制造质量控制方面的缺陷检测,考虑到工业摄像头角度和零部件表面缺陷特征相对固定的特点,提出一种基于注意力机制的YOLO缺陷检测算法。围绕提升算法注意力,首先采用CZS算法,把图像上的缺陷区域剪切、缩放和拼接成新图像,使注意力集中于缺陷相关区域;然后采用裁减主干网络算法,裁减掉原版YOLOv3主干网络中无用的检测尺度层;最后使用数据增强算法增加训练样本量。实验案例结果表明:该算法检测精度99.2%,单帧图像检测时间0.01 s,性能均优于原版YOLOv3;该算法在固定摄像头场景下具有一定先进性,3项提升注意力的策略使算法训练精度收敛的更快、检测速度更快、检测性能更稳定。

In this study,considering the industrial camera angle and the relatively fixed defect features on the part surface of parts manufacturing quality control,an attention mechanism based YOLO defect inspection algorithm is proposed.First,CZS algorithm is used to cut,zoom and splice the defect area on the image into a new image to focus on the defect related area;Then the algorithm of pruning the backbone network is used to prune the useless detection scale layer in the original YOLOv3 backbone network;Finally,the data enhancement algorithm is used to increase the training samples.The experimental results show that the detection accuracy of this algorithm is 99.2%,the detection time of a single frame image is 0.01 s,and the performance is better than the original YOLOv3;The algorithm is progressiveness in the fixed camera scene,and three strategies to improve attention make the algorithm converge faster in training accuracy and more stable in detection performance.