[发明专利]姿态获取方法、关键点坐标定位模型的训练方法和装置

说明书

本申请提供了一种姿态获取方法、关键点坐标定位模型的训练方法和装置。所述方法包括：获取目标视频；对于目标视频中的第i个图像帧，获取第i个图像帧的热度图，i为正整数；根据第i个图像帧的热度图以及第i个图像帧的邻近图像帧的热度图，得到第i个图像帧中关键点的三维坐标；根据第i个图像帧中关键点的三维坐标获取目标对象在第i个图像帧中的目标姿态。相比于相关技术中，本申请实施例提供的技术方案，可以有效克服目标对象在不同的图片中大小和姿态不同造成的预测不准确的问题，从而提升了预测的关键点位置的准确性，进一步提升了获取的姿态的准确性。

技术领域

本申请实施例涉及人工智能技术领域，特别涉及一种姿态获取方法、关键点坐标定位模型的训练方法和装置。

背景技术

随着计算机技术的发展，游戏、动画等的制作不再依赖于人工手绘图纸，可以借助于某些设备从图片中获取一些特定的姿态，基于获取到姿态进行制作。

在相关技术中，获取姿态获取可以包括如下步骤：首先，获取包括目标对象(如人体)的图片；然后，对图片中的目标对象进行检测，获取到包括目标对象的区域；之后，基于目标对象的区域预测目标对象的关键点的二维坐标；最后，将关键点的二维坐标输入至姿态获取模型，获取到该图片中目标对象的姿态。

由于目标对象在不同的图片中大小和姿态可能不同，若采用相关技术中，基于图片中预测的关键点的二维坐标获取姿态，由于预测的关键点位置不准确，进而导致最终获取的姿态不准确。

发明内容

本申请实施例提供了一种姿态获取方法、关键点坐标定位模型的训练方法和装置，可用于解决相关技术中，由于预测的关键点位置不准确，进而导致最终获取的姿态不准确的问题。所述技术方案如下：

一方面，本申请实施例提供了一种姿态获取方法，所述方法包括：

获取目标视频；

对于所述目标视频中的第i个图像帧，获取所述第i个图像帧的热度图，所述第i个图像帧的热度图包括所述第i个图像帧中目标对象的各个关键点的热度图，所述热度图中像素点的值用于指示所述像素点为关键点的概率，所述i为正整数；

根据所述第i个图像帧的热度图以及所述第i个图像帧的邻近图像帧的热度图，得到所述第i个图像帧中关键点的三维坐标；

根据所述第i个图像帧中关键点的三维坐标获取所述目标对象在所述第i个图像帧中的目标姿态。

另一方面，本申请实施例提供了一种关键点坐标定位模型的训练方法，所述方法包括：

获取至少一组训练样本，每组训练样本包括样本视频、所述样本视频中样本对象的关键点的真实二维坐标和所述关键点的真实三维坐标；

在所述样本视频中对所述样本对象进行检测跟踪；

获取所述样本视频中各个样本图像帧的热度图，所述样本图像帧的热度图包括所述样本图像帧中所述样本对象的各个关键点的热度图，所述热度图中像素点的像素值用于指示所述像素点为关键点的概率；

采用各个所述样本图像帧的热度图、所述关键点的真实二维坐标和所述关键点的真实三维坐标，对关键点坐标定位模型进行训练，所述关键点坐标定位模型用于根据目标样本图像帧的热度图和所述目标样本图像帧的邻近样本图像帧的热度图，确定所述目标样本图像帧中关键点的预测三维坐标；

当满足停止训练条件时，停止对所述关键点坐标定位模型的训练，得到完成训练的关键点坐标定位模型。

又一方面，本申请实施例提供了一种姿态获取装置，所述装置包括：

视频获取模块，用于获取目标视频；