[发明专利]人体关键点提取方法、装置、可读存储介质及设备

说明书

本公开涉及一种人体关键点提取方法、装置、可读存储介质及设备，所述方法包括：接收包含人体图像部分的目标图像；提取所述目标图像中的目标人体特征图；将所述目标人体特征图输入至关键点提取模型，以获得人体关键点热图；其中，所述关键点提取模型包括第一子模型和第二子模型，所述第一子模型用于根据所述目标人体特征图获得人体关键点热图，所述第二子模型用于根据所述目标人体特征图获得关键点回归坐标信息；所述关键点提取模型的损失值是根据所述第一子模型的损失值和所述第二子模型的损失值确定的。通过上述技术方案，可以有效提高提取出的人体关键点的准确度，提升用户使用体验。

技术领域

本公开涉及图像处理领域，具体地，涉及一种人体关键点提取方法、装置、可读存储介质及设备。

背景技术

在进行图像关键点提取时，现有技术中，通常是通过卷积神经网络提取图像的关键点，将标记的图像进行统一训练。然而，当图像中包含过多的人物时，若两个人物距离较近时，在确定其中某一人物的关键点时，通过上述方式难以对该不同人物的关键点进行区分。

发明内容

提供该发明内容部分以便以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。该发明内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。

第一方面，本公开提供一种人体关键点提取方法，所述方法包括：

接收包含人体图像部分的目标图像；

提取所述目标图像中的目标人体特征图；

将所述目标人体特征图输入至关键点提取模型，以获得人体关键点热图；

其中，所述关键点提取模型包括第一子模型和第二子模型，所述第一子模型用于根据所述目标人体特征图获得人体关键点热图，所述第二子模型用于根据所述目标人体特征图获得关键点回归坐标信息；所述关键点提取模型的损失值是根据所述第一子模型的损失值和所述第二子模型的损失值确定的。

第二方面，本公开提供一种人体关键点提取装置，所述装置包括：

接收模块，用于接收包含人体图像部分的目标图像；

第一提取模块，用于提取所述目标图像中的目标人体特征图；

第一输入模块，用于将所述目标人体特征图输入至关键点提取模型，以获得人体关键点热图；

其中，所述关键点提取模型包括第一子模型和第二子模型，所述第一子模型用于根据所述目标人体特征图获得人体关键点热图，所述第二子模型用于根据所述目标人体特征图获得关键点回归坐标信息；所述关键点提取模型的损失值是根据所述第一子模型的损失值和所述第二子模型的损失值确定的。

第三方面，本公开提供一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现第一方面中任一项所述方法的步骤。

第四方面，本公开提供一种电子设备，包括：

存储器，其上存储有计算机程序；

处理器，用于执行所述存储器中的所述计算机程序，以实现第一方面中任一项所述方法的步骤。