[发明专利]一种图像生成方法、装置及设备

说明书

本发明实施例提供了一种图像生成方法、装置及设备；该方法包括：获取训练样本，并确定当前预设训练阶段；当前预设训练阶段为至少两个预设训练阶段中的任一预设训练阶段；确定当前预设训练阶段的待训练模型，得到当前待训练图像生成模型；设置当前预设训练阶段中至少两个预设部位分别对应的权重值，得到当前权重张量；利用训练样本和当前权重张量，持续对当前待训练图像生成模型进行训练，直到满足当前预设训练条件为止，得到当前目标图像生成模型；将当前目标图像生成模型作为下一预设训练阶段的待训练模型进行阶段训练，直到完成至少两个预设训练阶段的训练为止，得到目标图像生成模型。通过本发明实施例，能够提升图像生成效果。

技术领域

本发明涉及人工智能领域中的图像处理技术，尤其涉及一种图像生成方法、装置及设备。

背景技术

图像生成指对待生成图像进行补全或变换风格等处理，以得到目标图像的过程；比如，将一个人脸中缺失五官的待生成图像进行五官补全得到具有完整五官的目标图像的过程，又比如，将一个素描风格的待生成图像变换为真实风格的目标图像的过程。通过图像生成，提升了图像处理的多样性。

一般来说，进行图像生成时，通常利用训练好的图像生成模型对待生成图像进行图像生成，得到目标图像。然而，上述图像生成的过程中，由于图像生成模型在训练过程中，为了提升目标图像中一个特定部位的生成效果，通常对特定部位进行针对性训练，如此，导致目标图像中其他特定部位的生成效果降低；因此，图像生成效果差。

发明内容

本发明实施例提供一种图像生成方法、装置及设备，能够提升图像生成效果。

本发明实施例的技术方案是这样实现的：

本发明实施例提供一种图像生成方法，包括：

获取训练样本，并确定当前预设训练阶段；所述训练样本用于训练对待生成图像进行图像生成的模型，所述当前预设训练阶段为至少两个预设训练阶段中的任一预设训练阶段；

确定所述当前预设训练阶段的待训练模型，得到当前待训练图像生成模型；

设置所述当前预设训练阶段中至少两个预设部位分别对应的权重值，得到当前权重张量；

利用所述训练样本和所述当前权重张量，持续对所述当前待训练图像生成模型进行训练，直到满足当前预设训练条件为止，得到当前目标图像生成模型；

将所述当前目标图像生成模型作为下一预设训练阶段的待训练模型进行阶段训练，直到完成所述至少两个预设训练阶段的训练为止，得到目标图像生成模型；所述目标图像生成模型用于对所述待生成图像进行图像生成。

本发明实施例提供一种图像生成装置，包括：

阶段确定模块，用于获取训练样本，并确定当前预设训练阶段；所述训练样本用于训练对待生成图像进行图像生成的模型，所述当前预设训练阶段为至少两个预设训练阶段中的任一预设训练阶段；

模型确定模块，用于确定所述当前预设训练阶段的待训练模型，得到当前待训练图像生成模型；

权重设置模块，用于设置所述当前预设训练阶段中至少两个预设部位分别对应的权重值，得到当前权重张量；

模型训练模块，用于利用所述训练样本和所述当前权重张量，持续对所述当前待训练图像生成模型进行训练，直到满足当前预设训练条件为止，得到当前目标图像生成模型；

所述模型训练模块，还用于将所述当前目标图像生成模型作为下一预设训练阶段的待训练模型进行阶段训练，直到完成所述至少两个预设训练阶段的训练为止，得到目标图像生成模型；所述目标图像生成模型用于对所述待生成图像进行图像生成。

本发明实施例提供一种图像生成设备，包括：

存储器，用于存储可执行指令；