具有对抗性训练的伪异常样本表面缺陷检测方法及设备技术

本发明专利技术针对工业产品表面缺陷在实际应用中“样本不平衡”问题，以及由于非显著缺陷和不同缺陷之间的相似性，高效定位和分类表面缺陷仍然是一个挑战的问题，提出了一种具有对抗性训练的伪异常样本表面缺陷检测方法及设备，在不需要额外异常样本学习的基础上能够适应检测数据集，从而达到工业产品表面缺陷的有效检测。方法中混合对象检测器，特征编解码器和分类器。对象检测器能够在正式检测之前先筛选一轮异常区域达到整个模型的高效检测能力。特征编解码器具有伪异常样本的对抗性训练，能够使得模型在不需要真实的异常样本的情况下完成数据集正常和异常的学习，减少实际应用中“样本不平衡”对检测结果的影响。对检测结果的影响。对检测结果的影响。

【技术实现步骤摘要】
具有对抗性训练的伪异常样本表面缺陷检测方法及设备


[0001]本专利技术涉及计算机视觉与缺陷检测
，特别涉及一种具有对抗性训练的伪异常样本表面缺陷检测方法及设备。

技术介绍

[0002]物体表面缺陷检测是工业自动化生产不可或缺的一个环节，研究物体表面缺陷检测技术具有很强的现实意义。基于传统机器视觉的物体表面缺陷检测方法一般流程包括：图像处理、特征提取、目标分类。实际工作中，在获取图像数据后，通常采用图像增强和图像分割等处理技术预先分离出缺陷区域，然后再根据缺陷目标人为设计特征规则，手工提取缺陷区域的特征，最后应用一些分类算法对缺陷特征进行分类。
[0003]但传统的机器视觉缺陷检测技术在实际应用中遇到了许多问题和挑战，特别是在那些用于区分缺陷和非缺陷的传统图像特征是基于经验手动设计的年代。传统图像特征提取算子的特征通常处于低水平。在诸如照明变化、透视失真、遮挡、对象变形等复杂场景变化的情况下，提取的特征通常不够鲁棒，无法处理它们，因此许多算法在实际环境中不适用。而深度学习的快速发展使其在缺陷检测领域得到了越来越广泛的应用。
...

【技术保护点】

【技术特征摘要】
1.一种具有对抗性训练的伪异常样本表面缺陷检测方法，其特征在于，包括以下步骤：S1：构建一个对象检测器，所述对象检测器包括：特征提取网络和RPN；S2：将正常对象送入对象检测器，得到n个目标子样本；S3：构建特征编解码器，所述特征编解码器由CAE组成，CAE包括：编码器、第一解码器和第二解码器；S4：将对象检测器得到的n个目标子样本中的任一目标子样本送入编码器，将与表面缺陷检测数据集无关的数据作为伪异常样本送入编码器，经过第一解码器进行正常样本的训练，重构正常对象，经过第二解码器进行伪异常样本的对抗性训练，解码伪异常对象；S5：继续将剩余的n
‑
1个目标子样本重复S4，直至一个对象结束；同时根据各损失更新网络参数，完成编码器的训练；S6：计算各目标子样本和其重构之后的绝对差，计算伪异常样本与其解码结果的绝对差；S7：构建分类器并训练分类器；S8：测试阶段，去掉第二解码器，将测试样本送入已经训练完成的对象检测器、CAE和分类器，得出表面缺陷检测结果。2.根据权利要求1所述的伪异常样本表面缺陷检测方法，其特征在于，步骤S1中，所述特征提取网络采用Resnet
‑
50，在ImageNet上预训练后再用能收集到的工业产品表面缺陷再一次训练来提高精度；RPN用于得到建议框，将每个建议框裁取得到疑似目标区域每一个区域作为一个目标子样本，在训练阶段，采用随机获取建议框的形式。3.根据权利要求1所述的伪异常样本表面缺陷检测方法，其特征在于，步骤S3中，每个编码器由三个卷积层组成，每个卷积层之后是最大池化层，滤波器大小为2
×
2，步长为2；卷积层由3
×
3的滤波器组成；每个卷积层之后都有ReLU作为激活函数；前两个卷积层由32个过滤器组成，而第三个卷积层由16个过滤器组成，潜在表示由16个大小为8
×
8的激活图组成；每个解码器从上采样层开始，将激活图的空间支持增加2倍；上采样操作基于最近邻插值；上采样后，应用了一个带有16个3
×
3滤波器的卷积层；第一个上采样层和卷积层之后是另两个上采样层与卷积层，最后一个卷积层由单个卷积滤波器构成。4.根据权利要求1所述的伪异常样本表面缺陷检测方法，其特征在于，步骤S4中，对于重构正常对象的第一解码器d，采用MSE损失函数作为损失函数L
d
：其中，h、w分别为输入正常样本的高和宽并且h＝w＝64，x
ij
为以h为行w为列的第i行第j列的正常样本的灰度图数值，为x
ij
通过解码器之后得到的数值，θ
e
...

【专利技术属性】
技术研发人员：罗大鹏，黄罗琪，罗人立，洪世杰，李金生，王菲，蹇安安，杜浩文，
申请(专利权)人：中建三局智能技术有限公司罗人立，
类型：发明
国别省市：