一种基于权重图引导的布匹瑕疵图像生成方法

本发明涉及图像生成技术领域，公开了一种基于权重图引导的布匹瑕疵图像生成方法，包括采集带有瑕疵的布匹图片，经图像预处理为一组子图像然后依次输入到离线训练好的WMG‑GAN网络中生成新的布匹瑕疵图片。WMG‑GAN网络基于CycleGAN网络，对CycleGAN网络生成器的输出结果图像增加采用注意力机制的思路，将输出结果图像分为前景权重图和特征权重图；在CycleGAN网络判别器的每个中间层级的LeakyReLU函数后增加ConvNeXtV2模块。本发明的图像生成方法能够在无需配对样本的情况下生成布匹瑕疵图片，通过使用前景和特征权重图可以选择性地改变前景同时保留背景，从而实现更加真实的图像生成。