图像生成模型的训练方法和装置以及图像生成方法和装置

本申请公开了图像生成模型的训练方法和装置以及图像生成方法和装置，其中的训练方法包括：利用第一缺陷图像及粗略掩码，对主分支进行训练且屏蔽副分支处理；完成主分支训练后，利用第二缺陷图像及粗略掩码和精细掩码对副分支进行训练；副分支进行训练时，将第二缺陷图像及精细掩码输入副分支，经过第二VAE编码器和第二UNET编码器的处理，得到第二UNET编码器输出；将第二缺陷图像及粗略掩码输入主分支，将第二UNET编码器输出输入第一UNET解码器，处理后获取UNET解码器输出的预测噪声，并基于预测噪声更新副分支的模型参数。应用本申请，训练生成的模型可以批量生成具有多样性的缺陷图像，同时能够增强缺陷的可控性。