论文部分内容阅读
在自动化工业检测中,通常需要在无异常图像上训练模型并对产品进行异常检测,这在计算机视觉领域中是一项重要并具有挑战性的任务.先用学生-教师网络训练学生回归教师的输出,再用学生与教师的输出差异定位异常,在异常分割领域取得了先进的效果,然而,该算法预测速度慢,且不能实现预测和异常检测同时进行.针对以上问题,提出了一种多学生-教师网络:使用多个学生网络联合回归教师网络的输出,选择学生与教师的输出在各维度上平方差的最小值作为差异值;用网络中间层信息代表图像各区域计算异常距离并进行异常分割,用最大异常分代表图片的异常程度进行异常检测.在MVTec AD上的实验表明,该算法预测一张图用时0.17 s,可以同时输出异常检测结果,其Image AUROC达到91.1％,Pixel AUROC达到94.5％.在现实场景拍照制作的墙面瓷砖数据集上,Image AUROC达到89.7％,Pixel AUROC达到89.1％.相比于原学生-教师网络,该方法可以快速完成异常分割与异常检测任务,具有更高的精度,在实际应用中取得了较好的效果.“,”In automated industrial inspection,it is often necessary to train models on anomaly-free images and perform anomaly detection on products,which is also an important and challenging task in computer vision.The student-teacher network trains students to regress the output of the teacher,and uses the difference between the output of the student network and the pre-trained teacher network to locate anomalies,which has achieved advanced results in the field of abnormal segmentation. However,it is slow to predict a picture,and no anomaly detection is performed.A multi-student teacher network is proposed, which uses multiple student networks to jointly regress the output of the teacher network,and the minimum square difference between the output of students and teachers in each dimension is selected as the difference value.The information in the middle layer of the network is used to represent each area of the image and calculate the anomaly distance for anomaly segmentation, and the maximum abnormal score is used to represent the abnormal degree of the image for abnormal detection.Experiments results on MVTec anomaly detection show that the algorithm predicts a picture in 0.17 s and can output anomaly detection results at the same time,with image AUROC reaching 91.1％ and Pixel AUROC reaching 94.5％.On the wall tile dataset produced by taking pictures of real scenes,image AUROC reached 89.7％,and Pixel AUROC reached 89.1％.Compared with the original student-teacher network,the proposed method can quickly complete anomaly segmentation and anomaly detection tasks at the same time with better accuracy,and it also has better results in real applications.