[发明专利]姿势优化模型的训练方法、使用方法、装置及处理设备

权利要求书

1.一种姿势优化模型的训练方法，其特征在于，包括：

获取训练样本，所述训练样本包括对象的图像、初始姿势信息和姿势评价信息；

确定姿势优化模型的姿势评价损失函数和姿势改变损失函数；其中，所述姿势评价损失函数表示所述对象的姿势优美程度；所述姿势改变损失函数表示所述对象的姿势改变量；

将所述姿势评价损失函数和所述姿势改变损失函数按照预设的权重系数求和，得到总损失函数；

使用所述初始姿势信息、所述图像和所述姿势评价信息对所述姿势优化模型进行训练，直至所述总损失函数收敛，将所述总损失函数收敛时对应的参数作为所述姿势优化模型的参数。

2.根据权利要求1所述的方法，其特征在于，所述获取训练样本的步骤，包括：

通过关键点检测模型对所述对象的图像进行关键点检测，获取所述初始姿势信息；接收用户输入的姿势评价信息。

3.根据权利要求1所述的方法，其特征在于，所述姿势优化模型包括：姿势评价模型和姿势改变模型；所述确定姿势优化模型的姿势评价损失函数和姿势改变损失函数的步骤，包括：

使用所述初始姿势信息和所述图像训练所述姿势改变模型，将所述姿势改变模型的输出的范数作为姿势改变损失函数；

使用所述初始姿势信息、优化姿势信息、所述图像和所述姿势评价信息训练所述姿势评价模型，将所述姿势评价模型的输出作为姿势评价损失函数；所述优化姿势信息为所述初始姿势信息与对应的所述姿势改变模型输出的和。

4.一种姿势优化模型的使用方法，其特征在于，应用于终端，所述终端包括权利要求1-3任一项所述姿势优化模型的训练方法得到的姿势优化模型，所述方法包括：

获取对象的实时图像及初始姿势信息；

将所述实时图像和所述初始姿势信息输入所述姿势优化模型，以使所述姿势优化模型输出姿势优化信息；

根据所述姿势优化信息确定姿势改变提示信息；

输出所述姿势改变提示信息。

5.根据权利要求4所述的方法，其特征在于，所述根据所述姿势优化信息确定姿势改变提示信息的步骤，包括：

根据所述姿势优化信息确定所述对象的关键点改变信息；

将所述关键点改变信息作为所述姿势改变提示信息。

6.一种姿势优化模型的训练装置，其特征在于，包括：

第一获取模块，用于获取训练样本，所述训练样本包括对象的图像、初始姿势信息和姿势评价信息；

第一确定模块，用于确定姿势优化模型的姿势评价损失函数和姿势改变损失函数；其中，所述姿势评价损失函数表示所述对象的姿势优美程度；所述姿势改变损失函数表示所述对象的姿势改变量；

第二确定模块，用于将所述姿势评价损失函数和所述姿势改变损失函数按照预设的权重系数求和，得到总损失函数；

模型训练模块，用于使用所述初始姿势信息、所述图像和所述姿势评价信息对所述姿势优化模型进行训练，直至所述总损失函数收敛，将所述总损失函数收敛时对应的参数作为所述姿势优化模型的参数。

7.一种姿势优化模型的使用装置，其特征在于，应用于终端，所述终端包括权利要求1-3任一项所述姿势优化模型的训练方法得到的姿势优化模型，所述装置包括：

第二获取模块，用于获取对象的实时图像及初始姿势信息；

优化模块，用于将所述实时图像和所述初始姿势信息输入所述姿势优化模型，以使所述姿势优化模型输出姿势优化信息；

提示确定模块，用于根据所述姿势优化信息确定姿势改变提示信息；

输出模块，用于输出所述姿势改变提示信息。

8.一种处理设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现上述权利要求1至5中任一项所述的方法的步骤。

9.一种具有处理器可执行的程序代码的计算机可读介质，其特征在于，所述程序代码使所述处理器执行上述权利要求1至5中任一所述方法。