基于改进的tiny-YOLOv3网络的表面缺陷检测研究

Research on Surface Defect Detection Based on Improved tiny-YOLOv3 Network

表面缺陷自动化检测在社会各个行业有广泛应用前景,可以大幅度提升效率。基于卷积神经网络架构的目标检测模型是自动化表面缺陷检测与识别的重要方法。折中检测速度与精确度,选择tiny-YOLOv3网络作为表面缺陷检测的模型。将视觉注意力机制引入tiny-YOLOv3网络结构并比较不同类别注意力机制在网络不同位置对于模型表现的影响,从而提出一种对于原网络改进的方法。改进的tiny-YOLOv3网络结构在表面缺陷数据集上测试结果较原始tiny-YOLOv3网络在mAP值上提升2.3%。

Automatic detection of surface defects has a wide application prospect in various industries of society,which can greatly improve the efficiency.The target detection model based on convolutional neural network architecture is an important method for automatic surface defect detection and recognition.To compromise the detection speed and accuracy,choose tiny-YOLOv3 network as the model for surface defect detection.The visual attention mechanism is introduced into the tiny-YOLOv3 network structure and the influence of different types of attention mechanisms in different network positions on the performance of the model are compared,then a method to improve the original network is proposed.The testing results of the improved tiny-YOLOv3 network structure in the surface defect dataset show that the mAP value is 2.3% higher than that of the original tiny-YOLOv3 network.