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

权利要求书

1.一种人体关键点提取方法，其特征在于，所述方法包括：

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

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

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

其中，所述关键点提取模型包括第一子模型和第二子模型，所述第一子模型用于根据所述目标人体特征图获得人体关键点热图，所述第二子模型用于根据所述目标人体特征图获得关键点回归坐标信息，其中，所述关键点回归坐标信息用于对所述人体关键点热图进行位置约束；所述关键点提取模型的损失值是根据所述第一子模型的损失值和所述第二子模型的损失值确定的，所述关键点提取模型的训练图像中标记有关键点坐标信息，所述第二子模型的损失值为根据所述训练图像提取出的历史人体特征图的关键点回归坐标信息和所述关键点坐标信息确定出的各个关键点对应的相似度和1之间的均方差。

2.根据权利要求1所述的方法，其特征在于，所述关键点提取模型通过以下方式进行训练：

提取训练图像中的历史人体特征图，所述训练图像中标记有关键点得分图和关键点坐标信息；

将所述历史人体特征图输入所述关键点提取模型，获得所述历史人体特征图的人体关键点热图和关键点回归坐标信息；

根据所述历史人体特征图的人体关键点热图和所述关键点得分图，确定所述第一子模型的损失值；

根据所述历史人体特征图的关键点回归坐标信息和所述关键点坐标信息，确定所述第二子模型的损失值；

根据所述第一子模型的损失值和所述第二子模型的损失值，确定所述关键点提取模型的损失值；

若根据所述关键点提取模型的损失值确定更新所述关键点提取模型，则根据所述关键点提取 模型的损失值更新所述关键点提取模型，并重新返回提取训练图像中的历史人体特征图的步骤，直至根据更新后的关键点模型的损失值确定结束更新所述关键点提取模型，获得训练后的关键点提取模型。

3.根据权利要求2所述的方法，其特征在于，所述第一子模型的损失值为所述历史人体特征图的人体关键点热图和所述关键点得分图之间的均方差。

4.根据权利要求1-3中任一项所述的方法，其特征在于，所述提取所述目标图像中的目标人体特征图，包括：

提取所述目标图像的人体图像部分对应的第一图像；

将所述第一图像对应的分辨率调整至预设分辨率，获得第二图像，并根据所述第二图像确定所述目标图像中的所述目标人体特征图。

5.一种人体关键点提取装置，其特征在于，所述装置包括：

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

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

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

其中，所述关键点提取模型包括第一子模型和第二子模型，所述第一子模型用于根据所述目标人体特征图获得人体关键点热图，所述第二子模型用于根据所述目标人体特征图获得关键点回归坐标信息，其中，所述关键点回归坐标信息用于对所述人体关键点热图进行位置约束；所述关键点提取模型的损失值是根据所述第一子模型的损失值和所述第二子模型的损失值确定的，所述关键点提取模型的训练图像中标记有关键点坐标信息，所述第二子模型的损失值为根据所述训练图像提取出的历史人体特征图的关键点回归坐标信息和所述关键点坐标信息确定出的各个关键点对应的相似度和1之间的均方差。