[发明专利]生成器的训练方法及装置、存储介质和电子装置

说明书

本发明公开了一种生成器的训练方法及装置、存储介质和电子装置，可应用于图像处理领域。包括：使用上一轮训练得到的风格图像生成器将第一真实图像转换成第二风格图像，使用上一轮训练得到的真实图像生成器将第二风格图像转换成第三真实图像；使用上一轮训练得到的真实图像生成器将第一风格图像转换成第二真实图像，并通过上一轮训练得到的风格图像生成器将所述第二真实图像转换成第三风格图像；确定预设的对抗损失函数在本轮训练时的取值；根据对抗损失函数在本轮训练时的取值，对上一轮训练得到的风格图像生成器和上一轮训练得到的真实图像生成器中的参数进行本轮训练的调整。本发明解决了图像风格化的准确率较低的技术问题。

技术领域

本发明涉及计算机领域，具体而言，涉及一种生成器的训练方法及装置、存储介质和电子装置。

背景技术

相关技术中，图像风格化可应用在娱乐、社交、内容创作等应用场景。图像风格化是目前数据图像处理的一个研究热点。

现有的传统图像翻译方法，使用真实图像和风格化的图像对神经网络模型进行训练，得到图像风格化模型。训练完成后可以将输入的真实图像进行风格化，得到渲染后的图像。但是目前的神经网络模型需要使用成对的真实图像和风格化图像进行训练。

以将人物卡通化为例，现有技术中需要使用同一个人物的真实图像和卡通画图像对神经网络模型进行训练。而在实际应用中，真实人像获取难度较低，而制作对应的卡通人像成本很高，无法对数据集中的每个真实人像一一制作卡通人像。而少量的训练数据导致模型的准确率较低，这限制了传统图像翻译方法在卡通渲染任务上的应用。

针对相关技术中，图像风格化的准确率较低的问题，目前尚未存在有效的解决方案。

发明内容

本发明实施例提供了一种生成器的训练方法及装置、存储介质和电子装置，以至少解决图像风格化的准确率较低的技术问题。

根据本发明实施例的一个方面，提供了一种生成器的训练方法，包括：使用上一轮训练得到的风格图像生成器将第一真实图像转换成第二风格图像，并使用上一轮训练得到的真实图像生成器将所述第二风格图像转换成第三真实图像；使用所述上一轮训练得到的真实图像生成器将第一风格图像转换成第二真实图像，并通过所述上一轮训练得到的风格图像生成器将所述第二真实图像转换成第三风格图像；通过所述第一真实图像和所述第一风格图像、所述第二真实图像和所述第二风格图像，以及所述第三真实图像和所述第三风格图像，确定预设的对抗损失函数在本轮训练时的取值；根据所述对抗损失函数在本轮训练时的取值，对所述上一轮训练得到的风格图像生成器和所述上一轮训练得到的真实图像生成器中的参数进行本轮训练的调整，以获取所述本轮训练得到的风格图像生成器和所述本轮训练得到的真实图像生成器，其中，所述风格图像生成器被设置为经过多轮训练后成为目标风格图像生成器，所述目标风格图像生成器用于将输入的真实图像转换为风格图像。

根据本发明实施例的另一个方面，提供了一种生成器的训练方法，包括：获取终端设备对目标对象进行拍摄所得到的目标真实图像；将所述目标真实图像输入目标风格图像生成器，得到所述目标真实图像转换成的目标风格图像，其中，所述目标风格图像生成器是对风格图像生成器经过多轮训练后得到的，所述对所述风格图像生成器经过多轮训练包括：使用上一轮训练得到的风格图像生成器将第一真实图像转换成第二风格图像，并使用上一轮训练得到的真实图像生成器将所述第二风格图像转换成第三真实图像；使用所述上一轮训练得到的真实图像生成器将第一风格图像转换成第二真实图像，并通过所述上一轮训练得到的风格图像生成器将所述第二真实图像转换成第三风格图像；通过所述第一真实图像和所述第一风格图像、所述第二真实图像和所述第二风格图像，以及所述第三真实图像和所述第三风格图像，确定预设的对抗损失函数在本轮训练时的取值；根据所述对抗损失函数在本轮训练时的取值，对所述上一轮训练得到的风格图像生成器和所述上一轮训练得到的真实图像生成器中的参数进行本轮训练的调整，以获取所述本轮训练得到的风格图像生成器和所述本轮训练得到的真实图像生成器；显示所述目标风格图像。