Zhao Weidong, Chen Feng, Huang Hancheng, Li Dan, Cheng Wei
College of Electrical and Information Engineering, Anhui University of Technology, Maanshan 243000, China.
Comput Intell Neurosci. 2021 Mar 17;2021:5592878. doi: 10.1155/2021/5592878. eCollection 2021.
In recent years, more and more scholars devoted themselves to the research of the target detection algorithm due to the continuous development of deep learning. Among them, the detection and recognition of small and complex targets are still a problem to be solved. The authors of this article have understood the shortcomings of the deep learning detection algorithm in detecting small and complex defect targets and would like to share a new improved target detection algorithm in steel surface defect detection. The steel surface defects will affect the quality of steel seriously. We find that most of the current detection algorithms for NEU-DET dataset detection accuracy are low, so we choose to verify a steel surface defect detection algorithm based on machine vision on this dataset for the problem of defect detection in steel production. A series of improvement measures are carried out in the traditional Faster R-CNN algorithm, such as reconstructing the network structure of Faster R-CNN. Based on the small features of the target, we train the network with multiscale fusion. For the complex features of the target, we replace part of the conventional convolution network with a deformable convolution network. The experimental results show that the deep learning network model trained by the proposed method has good detection performance, and the mean average precision is 0.752, which is 0.128 higher than the original algorithm. Among them, the average precision of crazing, inclusion, patches, pitted surface, rolled in scale and scratches is 0.501, 0.791, 0.792, 0.874, 0.649, and 0.905, respectively. The detection method is able to identify small target defects on the steel surface effectively, which can provide a reference for the automatic detection of steel defects.
近年来,随着深度学习的不断发展,越来越多的学者致力于目标检测算法的研究。其中,小而复杂目标的检测与识别仍是有待解决的问题。本文作者了解到深度学习检测算法在检测小而复杂的缺陷目标方面存在的不足,在此分享一种用于钢表面缺陷检测的新的改进目标检测算法。钢表面缺陷会严重影响钢材质量。我们发现当前大多数用于NEU-DET数据集检测的算法检测精度较低,因此针对钢材生产中的缺陷检测问题,我们选择在此数据集上验证一种基于机器视觉的钢表面缺陷检测算法。在传统的Faster R-CNN算法上进行了一系列改进措施,比如重构Faster R-CNN的网络结构。基于目标的小特征,我们采用多尺度融合训练网络。针对目标的复杂特征,我们用可变形卷积网络替换部分传统卷积网络。实验结果表明,用该方法训练的深度学习网络模型具有良好的检测性能,平均精度均值为0.752,比原算法高0.128。其中,裂纹、夹杂、斑块、麻点、氧化皮压入和划痕的平均精度分别为0.501、0.79l、0.792、0.874、0.649和0.905。该检测方法能够有效识别钢表面的小目标缺陷,可为钢材缺陷的自动检测提供参考。