[发明专利]图像生成方法、装置、计算机设备和存储介质

权利要求书

1.一种图像生成方法，所述方法包括：

获取待处理对象图像和参考姿态数据；

对所述待处理对象图像进行姿态识别，得到所述待处理对象图像对应的对象骨架数据；

调用姿态迁移模型，所述姿态迁移模型是根据训练样本对应的对象误差和姿态误差训练得到的；训练步骤包括：确定多个姿态关键点对；确定所述训练样本的骨架训练结果；从所述骨架训练结果中获取与所述多个姿态关键点对各自对应的结果关键点对，组成多个关键点对组；生成至少一条姿态参考线；将所述关键点对组包括的所述姿态关键点对和所述结果关键点对，映射至所述姿态参考线，得到各自对应的姿态向量和训练结果向量；根据所述姿态向量和所述训练结果向量确定第一关键点对误差；获取所述姿态关键点对与极轴之间的姿态偏移量，以及所述结果关键点对与所述极轴之间的训练偏移量；根据所述姿态偏移量和所述训练偏移量确定第二关键点对误差；根据多个所述第一关键点对误差确定第一姿态误差，根据多个所述第二关键点对误差确定第二姿态误差；获取所述第一姿态误差和所述第二姿态误差各自对应的误差权重；根据所述误差权重对所述第一姿态误差和所述第二姿态误差进行加权运算，得到所述骨架训练结果对应的目标姿态误差；

将所述对象骨架数据和所述参考姿态数据输入至所述姿态迁移模型，根据所述姿态迁移模型进行姿态迁移处理，得到目标骨架数据；

根据所述目标骨架数据生成所述待处理对象图像对应的目标对象图像。

2.根据权利要求1所述的方法，其特征在于，所述根据所述姿态迁移模型进行姿态迁移处理，得到目标骨架数据包括：

对所述对象骨架数据和所述参考姿态数据进行特征提取，得到所述对象骨架数据对应的对象特征，以及所述参考姿态数据对应的姿态特征；

将所述对象特征与所述姿态特征进行特征融合，得到目标特征；

获取所述姿态迁移模型根据所述目标特征输出对应的目标骨架数据。

3.根据权利要求1所述的方法，其特征在于，所述方法还包括所述姿态迁移模型的训练步骤，所述训练步骤还包括：

获取训练样本，所述训练样本包括对象训练样本和姿态训练样本；

获取初始迁移模型，将所述对象训练样本和所述姿态训练样本输入所述初始迁移模型，得到所述训练样本对应的骨架训练结果；

根据所述骨架训练结果和所述对象训练样本确定对象误差；

根据所述对象误差和所述姿态误差，对所述初始迁移模型中的模型参数进行修正。

4.根据权利要求3所述的方法，其特征在于，所述姿态训练样本包括姿态对象对应的姿态骨架数据，所述姿态骨架数据包括所述姿态对象对应的多个姿态关键点；确定姿态关键点对包括：

根据所述姿态骨架数据中的任意两个姿态关键点进行组合，得到多个姿态关键点对。

5.根据权利要求1所述的方法，其特征在于，所述姿态偏移量包括所述姿态关键点对向量方向与所述极轴之间的夹角。