[发明专利]头部姿态识别方法、装置、设备及计算机存储介质

说明书

本申请实施例提供了一种头部姿态识别方法、装置、设备及计算机存储介质，该头部姿态识别方法包括获取目标图像中目标对象的目标头部特征点在二维像素坐标系下的目标二维坐标；基于头部特征点与预设基准模型中预设特征点的预设对应关系，确定与目标头部特征点对应的预设特征点在世界坐标系下的目标三维坐标；基于二维像素坐标系与世界坐标系的预设变换关系，根据目标二维坐标以及目标三维坐标计算目标对象的头部姿态角，得到目标头部姿态。根据本申请实施例，可以提高头部姿态识别的准确性。

技术领域

本申请属于图像识别领域，尤其涉及一种头部姿态识别方法、装置、设备及计算机存储介质。

背景技术

随着图像识别技术的发展，姿态识别成为其中一个重要的研究方向，在广告、动画、多媒体以及社交等多个领域都有较大的应用前景，而在这些场景中都需要进行头部姿态的识别。因此，如何准确的识别出人体头部的姿态，成为亟待解决的问题。

现有技术中，主要通过基于模板匹配的方法，将输入的二维图像与模板库中标注过的图像进行比较，从而匹配得到与输入的二维图像最为相似的视图及头部姿态。然而，以上方式存在匹配过程耗时较长，并且匹配出的姿态结果准确度不高的问题。

发明内容

本申请实施例提供一种头部姿态识别方法、装置、设备及计算机存储介质，能够解决现有技术中头部姿态识别结果准确率低的问题。

第一方面，本申请实施例提供一种头部姿态识别方法，该方法包括：

获取目标图像中目标对象的目标头部特征点在二维像素坐标系下的目标二维坐标；

基于头部特征点与预设基准模型中预设特征点的预设对应关系，确定与所述目标头部特征点对应的所述预设特征点在世界坐标系下的目标三维坐标；

基于所述二维像素坐标系与所述世界坐标系的预设变换关系，根据所述目标二维坐标以及所述目标三维坐标计算所述目标对象的头部姿态角，得到目标头部姿态。

在一种实施方式中，所述获取目标图像中目标对象的目标头部特征点在二维像素坐标系下的目标二维坐标，包括：

采集视频流数据中的所述目标图像；

识别所述目标图像中目标对象的目标头部特征点；

基于预设提取算法获取所述目标头部特征点在所述二维像素坐标系下的所述目标二维坐标。

在一种实施方式中，在所述基于头部特征点与预设基准模型中预设特征点的预设对应关系，确定与所述目标头部特征点对应的预设特征点在世界坐标系下的目标三维坐标之前，所述方法还包括：

获取第一图像中目标对象的头部特征点在二维像素坐标系下的二维坐标；

获取预设初始模型中的预设特征点在世界坐标系下的三维坐标；

将所述预设特征点在世界坐标系下的三维坐标投影至二维像素平面，得到所述预设特征点在所述二维像素坐标系下的投影二维坐标；

调整所述投影二维坐标，以使所述投影二维坐标与所述头部特征点在二维像素坐标系下的二维坐标重合，得到所述头部特征点与预设初始模型中的预设特征点的预设对应关系。

在一种实施方式中，所述调整所述投影二维坐标，以使所述投影二维坐标与所述头部特征点在二维像素坐标系下的二维坐标重合，得到所述头部特征点与预设初始模型中的预设特征点的预设对应关系，包括：

基于所述头部特征点与所述预设初始模型中所述预设特征点的预设对应关系和预设拟合模型，调整所述投影二维坐标，以使所述投影二维坐标与所述头部特征点在二维像素坐标系下的二维坐标重合，得到所述头部特征点与预设初始模型中的预设特征点的预设对应关系。

其中，所述预设拟合模型基于如下公式表征：