[发明专利]图像生成模型训练方法以及装置

说明书

本说明书实施例提供图像生成模型训练方法以及装置，其中所述图像生成模型训练方法包括：获取目标对象的至少两张第一样本图像以及第一样本基准图像；计算每张第一样本图像相对于所述第一样本基准图像的第一样本位姿变换矩阵；将所述每张第一样本图像进行深度估计，获得所述每张第一样本图像的第一样本深度图像；基于所述第一样本图像、所述第一样本位姿变换矩阵以及所述第一样本深度图像对初始图像生成模型进行模型训练，获得针对所述目标对象的三维图像的图像生成模型。

技术领域

本说明书实施例涉及计算机技术领域，特别涉及图像生成模型训练方法以及装置。

背景技术

随着计算机技术的发展，三维图像被广泛的应用到各个领域，比如，可以应用到身份识别领域、动画制作领域、以及特效制作领域等。然而用以构造三维图像的图像生成模型的制作需要用高精度的3D扫描仪进行对待构建的对象进行上百次的扫描，并通过复杂的后处理才能获得，制作成本在百万元级别，因此，需要提供更经济且有效的方案。

发明内容

有鉴于此，本说明书实施例提供了一种图像生成模型训练方法。本说明书一个或者多个实施例同时涉及一种图像生成模型训练装置，一种计算设备，以及一种计算机可读存储介质，以解决现有技术中存在的技术缺陷。

本说明书实施例的第一方面，提供了一种图像生成模型训练方法，包括：

获取目标对象的至少两张第一样本图像以及第一样本基准图像；

计算每张第一样本图像相对于所述第一样本基准图像的第一样本位姿变换矩阵；

将所述每张第一样本图像进行深度估计，获得所述每张第一样本图像的第一样本深度图像；

基于所述第一样本图像、所述第一样本位姿变换矩阵以及所述第一样本深度图像对初始图像生成模型进行模型训练，获得针对所述目标对象的三维图像的图像生成模型。

本说明书实施例的第二方面，提供了一种图像生成模型训练装置，包括：

获取模块，被配置为获取目标对象的至少两张第一样本图像以及第一样本基准图像；

计算模块，被配置为计算每张第一样本图像相对于所述第一样本基准图像的第一样本位姿变换矩阵；

深度估计模块，被配置为将所述每张第一样本图像进行深度估计，获得所述每张第一样本图像的第一样本深度图像；

训练模块，被配置为基于所述第一样本图像、所述第一样本位姿变换矩阵以及所述第一样本深度图像对初始图像生成模型进行模型训练，获得针对所述目标对象的三维图像的图像生成模型。

本说明书实施例的第三方面，提供了一种计算设备，包括：

存储器和处理器；

所述存储器用于存储计算机可执行指令，所述处理器用于执行所述计算机可执行指令：

获取目标对象的至少两张第一样本图像以及第一样本基准图像；

计算每张第一样本图像相对于所述第一样本基准图像的第一样本位姿变换矩阵；

将所述每张第一样本图像进行深度估计，获得所述每张第一样本图像的第一样本深度图像；

基于所述第一样本图像、所述第一样本位姿变换矩阵以及所述第一样本深度图像对初始图像生成模型进行模型训练，获得针对所述目标对象的三维图像的图像生成模型。

本说明书实施例的第四方面，提供了一种计算机可读存储介质，其存储有计算机可执行指令，该指令被处理器执行时实现所述图像生成模型训练方法的步骤。