基于手绘草图和图像样例约束的自然图像生成方法及装置

摘要

本发明公开了基于手绘草图和图像样例约束的自然图像生成方法及装置，方法包括：首先获取原始自然图像和类别信息，构建训练数据集；接着构建自然图像生成模型，所述自然图像生成模型包括生成器和多任务判别器，所述生成器用于在训练过程中生成自然图像，所述多任务判别器用于在训练过程中判断生成的自然图像是否真实，并判断生成的自然图像的所属类别；然后通过所述训练数据集对所述自然图像生成模型进行训练，调整所述自然图像生成模型的参数，得到训练好的目标模型；最后将目标图像样例和目标手绘草图输入所述目标模型，生成基于所述目标手绘草图和图像样例约束的自然图像。本发明提高了便捷性以及可控性，可广泛应用于图像处理技术领域。