CN116071302B 一种基于双流半掩码重建的瓶口缺陷检测方法 （西安电子科技大学）

一种基于双流半掩码重建的瓶口缺陷检测本发明公开了一种基于双流半掩码重建的是否存在瓶口缺陷。本发明通过训练缺陷检测2将输入的瓶口图像调整为预定尺寸，将尺寸调整后的图像分割为采用预设的掩码率对所述多个图像块进行随机掩码，并将可见所述编码器包括依次连接的多个第一Transformer结构单元，用于提取可见块的特征所述解码器包括依次连接的多个第二Transformer结构单元，用于将所述编码器的输所述第一归一化层用于对输入的可见图像块进行归一所述全连接层用于对归一化处理后的可见图像所述多头注意力子单元用于通过多个注意力头分别对分解的每个部分进所述第一融合子单元用于将所述拼接后的特征图与输入的可见图像块所述第二归一化层用于对组合后的特征图进行归一所述多层感知机包括两个全连接层以及设置在两个全连接层之间的非线性激活函数，所述第二融合子单元用于将所述第一融合子单元的输出结果与所述多层感知机的输39.根据权利要求1至8中任一项所述的基于双流45[0016]所述编码器包括依次连接的多个第一Transformer结构单元，用于提取可见块的[0017]所述解码器包括依次连接的多个第二Transformer结构单元，用于将所述编码器的输出与输入瓶口图像的掩码部分进行位置嵌入，并完成对输入瓶口图像掩码部分的重[0020]所述全连接层用于对归一化处理后的可见图像块进行线性变换，分解为三个部[0021]所述多头注意力子单元用于通过多个注意力头分别对分[0024]所述多层感知机包括两个全连接层以及设置在两个全连接层之间的非线性激活[0025]所述第二融合子单元用于将所述第一融合子单元的输出结果与所述多层感知机6[0041]S5.1：计算所述待检测瓶口图像与所述重建图像中对应像素点的像素差值绝对[0047]3、本发明考虑到基于生成对抗网络训练花费昂贵计算资源的不足和基于卷积神[0049]图1是本发明实施例提供的一种基于双流半掩码重建的瓶口缺陷检测方法的流程7[0051]图3是本发明实施例提供的一种基于双流半掩码重建的瓶口缺陷检测方法的处理8[0066]所述编码器包括依次连接的多个第一Transformer结构单元，用于提取可见块的的输出与输入瓶口图像的掩码部分进行位置嵌入，并完成对输入瓶口图像掩码部分的重部分，分别为查询QeR、键keRK、值所述多头注意力子单元用于通9后将多头自注意力处理和块嵌入组合在一起的特征图与多层感知机处理后的特征图组合[0088]在本实施例中，在训练阶段之后还包括对训练后的瓶口缺陷检测系统的测试过[0097]S5.1：计算所述待检测瓶口图像与所述重建图像中对应像素点的像素差值绝对结果的混淆矩阵(Confusionmatr用来评估所提出的缺陷检测方法的性能。使用AUC值，即受试者工作特性(Receiver条曲线横轴为真正例率(TruePositiveRate，TPR)，纵轴为假正例率(FalsePositive[0114]根据AUC值的定义可知，通过对ROC曲线下各部分的面积求和可得到AUC值，假设[0117]本发明实施例转变了传统手工筛选瓶口缺陷产品的过程[0121]参照图8，本实例使用德国MVTec公司收集的现实工业场景下Bottledatasets数[0123]步骤1，训练阶段，首先使用数据处理方法将无缺陷图像尺寸调整为224×224大[0127]本发明实施例考虑到基于生成对抗网络训练花费昂贵计算资源的不足和基于卷积神经网络的自动编码器模型表征能力不足，提出一种基于Transformer的自动编码器网像随机掩码50并且将VisionTransfomer(ViT)结构引入自动编码器，编码器用来提取[0130]本发明的又一实施例提供了一种存储介质，所述存储介所述计算机程序用于执行上述实施例中所述基于双流半掩码重建的瓶口缺陷检测方法的包括：U盘、移动硬盘、只读存储器(Read_OnlyMemory，ROM)、随机存取存储器(Random