[发明专利]一种人体关键点识别方法、装置及电子设备

说明书

本发明实施例提供了一种人体关键点识别方法、装置及电子设备，应用于图像处理技术领域。该人体关键点识别方法包括：获得待识别人体关键点的目标图像；获得所述目标图像的人像蒙版；利用预先训练的神经网络模型，以所述目标图像的人像蒙版作为先验信息，对所述目标图像进行关键点信息识别，得到所述目标图像中每个人体关键点的热度图；其中，所述神经网络模型为利用样本图像和所述样本图像中每个人体关键点的真值热度图，并以所述样本图像的人像蒙版作为先验信息，所训练的模型；基于所述目标图像中每个人体关键点的热度图，确定所述目标图像中每个人体关键点的坐标。通过本方案可以提高人体关键点的识别速率。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种人体关键点识别方法、装置及电子设备。

背景技术

人体关键点识别是动作识别、异常行为检测、安防等的实现基础，其主要用于从给定的图像中定位人体的身体关键部位，例如头部、颈部、肩部、手部等部位。

现有技术中，在识别人体关键点时，将待识别人体关键点的目标图像输入至预先训练完成的神经网络模型，得到目标图像中每个人体关键点的热度图heat map；进而，基于每个人体关键点的热度图，计算每个人体关键点的坐标。其中，任一人体关键点的热度图为该人体关键点可能存在的位置的概率分布图。

发明人在实现本发明的过程中发现，现有技术至少存在如下问题：

神经网络模型的处理速率较低，最终导致人体关键点的识别速率较低。

发明内容

本发明实施例的目的在于提供一种人体关键点识别方法、装置及电子设备，以实现提高人体关键点的识别速率的目的。具体技术方案如下：

第一方面，本发明实施例提供了一种人体关键点识别方法，包括：

获得待识别人体关键点的目标图像；

获得所述目标图像的人像蒙版；

利用预先训练的神经网络模型，以所述目标图像的人像蒙版作为先验信息，对所述目标图像进行关键点信息识别，得到所述目标图像中每个人体关键点的热度图；其中，所述神经网络模型为利用样本图像和所述样本图像中每个人体关键点的真值热度图，并以所述样本图像的人像蒙版作为先验信息，所训练的模型；

基于所述目标图像中每个人体关键点的热度图，确定所述目标图像中每个人体关键点的坐标。

可选地，所述神经网络模型的训练过程包括：

获得多个样本图像，以及，每个样本图像的人像蒙版和每个样本图像中每个人体关键点的坐标；

针对每个样本图像，利用该样本图像中每个人体关键点的坐标，生成该样本图像中每个人体关键点的真值热度图；

针对每个样本图像，以该样本图像的人像蒙版作为先验信息，利用该样本图像对训练中的所述神经网络模型进行训练，得到该样本图像中每个人体关键点的预测热度图；

基于每个样本图像中每个人体关键点的真值热度图与预测热度图的差异，计算综合损失值；

基于所述综合损失值，判断训练中的所述神经网络模型是否收敛，如果收敛，结束训练，得到训练完成的所述神经网络模型；否则，调整训练中的所述神经网络模型的网络参数，继续训练所述神经网络模型。

可选地，针对每个样本图像，以该样本图像的人像蒙版作为先验信息，利用该样本图像对训练中的所述神经网络模型进行训练，得到该样本图像中每个人体关键点的预测热度图的步骤，包括：

针对每个样本图像，将该样本图像和该样本图像的人像蒙版进行关于通道维度的矩阵拼接，得到该样本图像对应的第一拼接矩阵；