[发明专利]一种基于深度学习的工业零件缺陷检测方法

说明书

本发明公开了一种基于深度学习的工业零件缺陷检测方法，涉及工业质检领域，主要包括步骤：获取预设数量的工业零件原始图像及经过缺陷标记后的缺陷标注图；根据缺陷标注图获取卷积池化处理后的特征图，将特征图与池化阶段中各池化层的输出进行融合，并利用初始卷积核卷积得到分割网络；调整卷积核大小依次对分割网络进行训练；根据原始图像和缺陷标注图对相应分割网络的输出结果进行分类训练，得到分类网络；根据分割网络和分类网络对工业零件原始图像进行缺陷程度、缺陷位置和缺陷种类判断。本发明通过先后池化‑上采样‑融合的处理，将缺陷分割的问题转换为分类问题，利用卷积神经网络擅长分类的优势，实现工业零件的缺陷标注和分类的高效性。

技术领域

本发明涉及工业质检领域，具体涉及一种基于深度学习的工业零件缺陷检测方法。

背景技术

在工业生产过程中，最终对工件表面的缺陷检测是最重要的。最主要采用的方法人工目测检测的方法，但其需要对员工进行培训，而小部分工厂也会采用例如ResNet50深层卷积网络和FPN结构构建特征金字塔，寻找缺陷特征，采用RetinaNet构建缺陷检测头模块，采用DeepLab V3或者U-NET构建语义特征提取模块来寻找缺陷的标记图像，但是他们都忽略了最重要的OK/NG检测无法自定检测的严苛程度，亦或者是在采用单独的OK/NG分类模型进行检测了例如IMAGENET进行分类。

人工目视法针对各种复杂的场景都可以适用，但是长时间的目视会让检测员工产生眼神经疲劳，导致检测效率和质量的下降，加上个体的不同，每个员工对标准的把控不同，导致出场的产品参差不齐。所以需要有一种客观准确的方法来进行工业上的缺陷检测。一种方案是采用Deeplab v3构建语义特征提取模块，但是如何寻找合适的rate，在增大感受野解决网格效应的同时避免模型退化是一个需要多次实验才能得出结果的问题，其不能满足工业4.0的速度要求，并且Deeplab v3参数量巨大，训练时间和训练需要的前期数据需求大，后期检测时间也很长，不能很好满足工业流水线的实时性要求；还有一种方案是采用ResNet50，不过其模型的参数量依旧巨大，需要专业的gpu并联而且训练几天才能达到比较好的效果，虽然在普通电脑上可以采用retrain的方式但是retrain的模型特征不一定符合当前需要检测的缺陷种类；一种方案是使用RetinaNet，它在准确度上达到很高的水平，但是只依靠RetinaNet进行检测并不能在工业检测上满足好坏检测以及缺陷位置标记，其结果无法给出综合评判好坏的分值也无法实现缺陷的像素分割；一种方案是采用U-NET，该网络模型在医学影像图像上有较为优秀的结果，但是针对工业图像由于工况的复杂性而难以收敛，收敛之后也有很大的可能性出现过拟合现象。

发明内容

为解决上述问题，本发明提出利用一个包含分割和分类在内的融合神经网络，在有限的参数量不影响检测速度的前提下保证输出缺陷位置图像分辨率和清晰的缺陷边界，并且给出OK/NG的分值以供工厂质检人员自行控制检测严苛程度，一种基于深度学习的工业零件缺陷检测方法，包括步骤：

S1：获取预设数量的工业零件原始图像及经过缺陷标记后的缺陷标注图；

S2：根据缺陷标注图获取卷积池化处理后的特征图，将特征图与池化阶段中各池化层的输出进行融合，并利用初始卷积核卷积得到分割网络；

S3：调整卷积核大小依次对分割网络进行训练；

S4：根据原始图像和缺陷标注图对相应分割网络的输出结果进行分类训练，得到分类网络；

S5：根据分割网络和分类网络对工业零件原始图像进行缺陷程度、缺陷位置和缺陷种类判断。

进一步地，所述步骤S1中，缺陷标注图为根据缺陷的种类，在原始图像缺陷对应位置设置相应灰度值的掩膜图像。

进一步地，所述步骤S2中，池化处理包括第一池化层至第三池化层，各池化层为别有预设数量的卷积层，池化层之间通过最大池化下采样层连接。

进一步地，特征图与池化阶段的池化层融合的具体步骤为：