【技术实现步骤摘要】
一种结合梯度解耦与对比学习的小样本瑕疵检测方法
[0001]本专利技术属于机器视觉
，特别涉及一种结合梯度解耦与对比学习的小样本瑕疵检测方法，用于实现少量人工标注场景下对工业产品表面进行瑕疵检测。
技术介绍
[0002]随着现阶段工业生产技术的不断发展，对于工业产品质量的要求也不断提高。在一些高度自动化的工业制造场景中，对产品的良率要求特别高。但是由于加工、设计、器械故障等因素，生产出的产品表面常常会出现一些瑕疵损坏，导致生产成本的增加、资源的浪费，严重时甚至会对使用者的人身安全造成伤害。因此在产品生产后就需要及时进行瑕疵检测，以确认部件或产品表面有无存在异物、瑕疵以及缺陷。
[0003]解决工业缺陷检测小样本瑕疵的问题，通常主要有两种分别是工程路径和算法路径。其中工程路径有着两种常见的方法，一种是基于真实产品手动制造缺陷检测，另一种为基于真实图像手动仿真缺陷检测。基于真实产品手动制造缺陷检测成本较高，存在一定的操作难度，且产品若在检测时损坏则不可逆。而基于真实图像手动仿真缺陷检测则检测速度较慢、难度大，对于操...
【技术保护点】
【技术特征摘要】
1.一种结合梯度解耦与对比学习的小样本瑕疵检测方法，其特征在于，主要包括如下步骤：步骤1，使用大规模的目标检测数据集经过预训练得到目标检测模型，所述目标瑕疵检测模型包括骨干网络、特征金字塔网络、候选框提取网络、候选框预测网络和梯度解耦模块；步骤2，通过自适应锚点算法为瑕疵数据集对应生成多个锚点；步骤3，调整所述目标检测模型中的所述候选框预测网络，将所述候选框预测网络的候选框分类分支输出的数量设置为瑕疵类别的数量，并在所述候选框预测网络中添加对比分支；步骤4，将所述瑕疵数据集和生成的所述锚点输入至调整后的所述目标检测模型中进行训练得到瑕疵检测模型；步骤5，对于待检测的瑕疵图像输入至所述瑕疵检测模型中，对应输出瑕疵目标的位置及其类别。2.根据权利要求1所述的一种结合梯度解耦与对比学习的小样本瑕疵检测方法，其特征在于，步骤1中所述大规模的目标检测数据集经过预训练得到目标检测模型，具体包括如下步骤：步骤1.1，初始化目标检测模型使用的数据加载器，提取图像和真实标注；步骤1.2，通过所述骨干网络提取所述图像的多层特征图；步骤1.3，使用所述特征金字塔网络融合所述多层特征图的信息得到金字塔特征图；步骤1.4，利用所述梯度解耦模块对所述金字塔特征图进行两次仿射变换得到的两组仿射特征图；步骤1.5，将两组所述仿射特征图分别输入至所述候选框提取网络和候选框预测网络中，所述候选框提取网络提取出含有目标的候选框，使用所述候选框预测网络计算出候选框类别和候选框回归参数；步骤1.6，根据所述候选框类别和候选框回归参数计算网络的分类损失、回归损失和梯度；步骤1.7，使用所述梯度解耦模块对所述候选框提取网络和候选框预测网络进行梯度解耦操作；步骤1.8，判断达到最大训练轮次后结束输出所述目标检测模型。3.根据权利要求2所述的一种结合梯度解耦与对比学习的小样本瑕疵检测方法，其特征在于，步骤2中所述通过自适应锚点算法为瑕疵数据集对应生成多个锚点，其具体包括如下步骤：步骤2.1，从瑕疵数据集中所有样本的瑕疵标注中随机取k个作为当前聚类中心；步骤2.2，判断是否有没计算过的所述瑕疵标注，若有则计算与当前所述瑕疵标注距离最近的聚类中心，并将当前所述瑕疵标注添加到所属聚类中心的列表中；步骤2.3，判断是否计算完所有的所述瑕疵标注，若已完成则计算所有列表中所述瑕疵标注的均值作为新聚类中心；步骤2.4，将所述当前聚类中心与新聚类中心进行比较，若一致则输出当前聚类中心对应瑕疵数据集的锚点，若不一致则返回步骤2.2继续执行。
4.根据权利要求3所述的一种结合梯度解耦与对比学习的小样本瑕疵检测方法，其特征在于，步骤4中所述瑕疵数据集和所述多个锚点经过训练得到瑕疵检测模型，其训练的具体步骤如下：步骤4.1，初始化瑕疵检测模型使用的数据加载器，提取图像和真实标注；步骤4.2，通过所述骨干网络提取所述瑕疵图像的多层特征图；步骤4.3，使用所述特征金字塔网络融合所述多层瑕疵特征图的信息得到瑕疵金字塔特征图；步骤4.4，利用所述梯度解耦模块对所述瑕疵金字塔特征图进行两次仿射变换得到的两组瑕疵仿射特征图；步骤4.5，将两组所述瑕疵仿射特征图分别输入至所述候选框提取网络和候选框预测网络中，所述候选框提取网络提取出含有瑕疵的候选框，使用所述候选框预测网络计算出候选框类别和候选框回归参数；步骤4.6，根据所述候选框类别和候选框回归参数计算网络的分类损失、回归损失、对比损失和梯度；步骤4.7，使用所述梯度解耦模块对所述候选框提取网络和候选框预测网络进行梯度解耦操作；步骤4.8，判断达到...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。