本技术提供了基于跨域异常注入的零样本伪异常图像合成方法及装置,包括以下步骤:收集真实异常图像,构建真实跨域异常数据集;选择真实跨域异常数据集中任一异常图像作为源图像,并获取源图像的二进制掩码,所述源图像的二进制掩码表示源图像中异常块的区域;基于给定的目标域正常图像集,获取目标图像,提取目标图像的二进制掩码表示前景;基于源图像与目标图像的二进制掩码将所述源图像的异常块缩放,注入目标图像中的指定位置,合成伪异常图像。本发明能够保证缩放的异常块能够自然、快速地注入到目标域的目标图像中,合成逼真的伪异常图像;且无需经过长时间训练便能生成大量逼真的伪异常图像。
背景技术
工业图像异常检测(IAD)是一个具有巨大价值的研究课题。由于异常的性质,在现代工业领域中真正的异常(即特定域的异常)通常过于罕见而无法收集,如成熟生产线上的异常发生率可能在0.1%以下,这严重阻碍了IAD的发展。因此,零样本异常合成技术(ZSAS)成为了IDA的一项重要技术,ZSAS指能在没有特定域异常图像的情况下合成伪异常图像。一方面,现有技术中,ZSAS要么从自身裁剪图像块作为异常纹理来源,要么从现有与异常检测无关的数据集裁剪图像块作为异常来源;由于现有技术中无法用简洁的方法将已有的异常纹理无缝地融入特定域的正常图像,从而合成逼真的伪异常图像,同时,由于没有真实异常纹理的参考,因此,现有技术中无法得到逼真的伪异常图像。另一方面,现有技术中,比如基于扩散或基于GAN的模型异常生成方式,天然需要长时间训练,并且需要参考特定域的异常图像,无法做到ZSAS;现有基于模型ZSAS也需要长时间训练,不仅成本高,而且无法解决真实性问题。
因此,需要一种在不参考特定域异常图像的前提下,更高效地合成更逼真的异常图像的技术方案。
实现思路