一种基于生成对抗网络的纺织品瑕疵样本数字化生成方法

摘要

本发明涉及一种基于生成对抗网络的纺织品瑕疵样本数字化生成方法，包括如下步骤：获取织物无瑕疵图像与瑕疵图像，建立数据集；创建生成对抗网络；整体训练生成对抗网络；判别器生成的注意力图会和无瑕疵织物纹理图进行合成，指导生成器需要生成的织物瑕疵；进行掩码正则化；将图像送进生成对抗网络中，生成出判别器和人眼都无法判断真假的瑕疵图；生成器生成出设定数量的瑕疵织物图片，实现扩充原有数据集；将生成的瑕疵前景和掩膜合成到不同的织物纹理背景上，形成新的数据集。本发明通过生成瑕疵前景与无瑕疵纹理背景合成的方法，来实现织物瑕疵的迁移，解决瑕疵检测网络训练过程中样本较少和没有瑕疵样本的问题。