[发明专利]一种基于知识蒸馏的工业图像缺陷检测方法

说明书

本发明涉及图像处理技术领域，尤其涉及一种基于知识蒸馏的工业图像缺陷检测方法，利用正常图像集合训练教师网络和学生网络，两者生成的特征图之间的差异作为损失函数，仅更新学生网络参数，在损失不再下降的情况下停止训练，得到训练后的学生网络；输入待检测图像至教师网络和学生网络，教师网络和学生网络不同尺度的特征图之间的差异记做异常分数标准，将特征图上采样到输入图像大小，对应位置相乘作为图像上各个像素点最终异常分数，设定阈值，大于阈值的是异常区域。本发明通过学生网络去学习教师网络中提取OK图像特征的方式，当遇到NG图像时，学生网络和教师网络之间的差异性凸显，以此达到检测NG图像的目的。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种基于知识蒸馏的工业图像缺陷检测方法。

背景技术

随着近几年深度学习的蓬勃发展，越来越多的深度学习算法被应用到各行各业中。其中深度学习算法中的知识蒸馏算法对于工业图像缺陷检测中的应用具有重要的研究和应用价值。

工业上的图像缺陷检测问题呈现几个明显的特点:1.检测图片中大部分都为OK样本，缺陷图片占总体检测图片的比例很小，非常不利于有监督算法对于缺陷需要预先进行标注的需求2.工业图像上缺陷的出现位置和形态呈现随机性，比较难对缺陷的类别进行定义3.工业图像上的缺陷各个种类之间的数量差异很大4.工业图像上的缺陷的出现存在未知性，很多缺陷会随着设备使用的老化出现。

针对以上总结的目前工业图像缺陷检测的现状和问题，传统图像算法和有监督深度学习算法对这种场景的解决程度存在一定的局限性，最终的检测效果通常也比较难达到理想的效果。为此，我们提出一种基于知识蒸馏的工业图像缺陷检测方法。

发明内容

基于背景技术存在的技术问题，本发明提出了一种基于知识蒸馏的工业图像缺陷检测方法，不需要有监督算法中需要对数据集进行筛选和标注，减少了大量人工工作量，解决了传统图像算法和有监督深度学习算法对工业图像缺陷检测的解决程度存在一定局限性的问题。

本发明提供如下技术方案：一种基于知识蒸馏的工业图像缺陷检测方法，包括如下步骤：

S1、首先给出训练的不包含缺陷的正常图像集合{I₁，I₂，...，I_n}，对输入图像集合中的每张图片做预处理；构建教师网络和学生网络，训练过程中输入图像I_k，教师网络和学生网络各输出三个不同尺度的特征图，使用教师网络和学生网络特征图之间的差异作为损失函数，训练过程使用SGD方法，对学生网络的参数进行更新，教师网络的参数保持不变，在损失不再下降的情况下停止训练，得到训练后的学生网络；

S2、对输入待检测图像进行预处理，将预处理后的图像同时输入到教师网络和学生网络，分别得到三个不同尺度的特征图，教师网络和学生网络不同尺度的特征图之间的差异记做异常分数标准：

最终的异常分数计算如下：

其中上采样的方式采用双线性插值，将各个特征图上采样到输入图像尺寸大小，待检测图像上越异常的地方，得分越高，通过设定阈值thresh，将大于该thresh地方的图像区域认为是检测图像上的异常区域。

优选的，I_k∈R^{wxh xc}，w为输入图像宽度，h为输入图像高度，c为输入图像通道数；

教师网络输出三个不同尺度的特征图为

学生网络输出三个不同尺度的特征图为

优选的，所述损失函数定义如下：

即为每个特征尺度对应位置(i,j)上的距离之和。