基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法

文档序号:37348073发布日期:2024-03-18 18:25阅读:来源:国知局

技术特征:

1.基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,包括生成式预训练网络和异常分割网络,其特征在于;

2.根据权利要求1所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述s1的生成式预训练网络具体包括以下步骤:

3.根据权利要求2所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述s1.3中预训练网络需要在小领域内学习正常图像中各元素间的相对位置关系,所述s1.2的网格掩模的基础上还需要叠加随机位置和更大尺度的大掩模。

4.根据权利要求1所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述s2的异常模拟策略具体包括以下步骤:

5.根据权利要求4所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述全卷积网络的异常模拟策略分为结构异常模拟和逻辑异常模拟两个支路;

6.根据权利要求5所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述逻辑异常模拟中全卷积网络需要利用卷积神经网络提取的深层特征图进行建模,且全卷积网络还需要利用自注意力建模全局范围内所有patch间的稠密关系和利用图卷积建模跨领域的、远程patch间的稀疏关系。

7.根据权利要求6所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述稠密关系中,对于编码器输出的深层特征应用自注意力建模全局范围内所有patch间稠密关系的遵从公式为和z=wdh(x),其中wd表示稠密相似度矩阵、f(·)和g(·)分别表示特征转换和维度调整函数。

8.根据权利要求6所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述稀疏关系中,对于编码器输出的深层特征需要先将其转化为图结构g(v,e),其中v为节点集合,v={vi}i=1:hw且然后构建图中节点之间的相似度矩阵ws,再借助ws对每一个节点索引与其相似的k各节点建立连接关系,即得到e={eij}i=1:hw,j=1:k。

9.根据权利要求8所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述相似度矩阵ws的构建方法具体包括以下步骤:

10.根据权利要求1所述的基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,其特征在于:所述s3中的优化生成器和判别器分别通过以下公式完成:其中xm为生成器的输入,x为真实图像,最后生成器预训练网络的优化目标表示为:异常分割网络的优化目标表示为:其中α表示为不同损失函数间的平衡因子。


技术总结
本发明公开了基于自监督学习和自注意力图卷积的半监督工业图像异常检测方法,完成半监督工业图像的表面缺陷检测,本方法在训练过程中通过人工模拟的异常样本引入异常的伪先验知识给模型增加训练约束,模型通过对比和区分异常的伪先验知识,可以更好地总结何为正常,并缩小用于描述正常特征分布的超球体,本发明的有益效果是:缓解了基于特征表示的模型在迁移学习时面临的域适应问题;还通过推理被掩模区域的可能形式,编码器对图像中位置信息的敏感性得以提高;在解码部分,模型利用图卷积强调Patch之间的重点逻辑关系,进而增强模型对逻辑异常的检测能力。

技术研发人员:刘静,杨明辉,滕祥意,杨智伟
受保护的技术使用者:西安电子科技大学广州研究院
技术研发日:
技术公布日:2024/3/17
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1