[发明专利]一种人体关键点识别方法、装置、终端设备及存储介质

权利要求书

1.一种人体关键点识别方法，其特征在于，包括：

通过关键点检测模型检测人体图像中人体的N个关节位置处的关键点，获得所述关键点检测模型输出的N张热图和一张背景图；其中，N为大于或等于2的整数；

分割出每张所述热图中不同人体的关键点所在区域，并确定每张所述热图中不同人体的关键点所在区域的轮廓；

在所述关键点区域的轮廓内获取热图峰值；

根据每张所述热图中不同人体的关键点所在区域的轮廓内的热图峰值的坐标，获得每张所述热图中不同人体的关键点的坐标；

根据每个所述人体的关键点的坐标，绘制每个所述人体的关节特征图。

2.如权利要求1所述的方法，其特征在于，所述根据每张所述热图中不同人体的关键点所在区域的轮廓内的热图峰值的坐标，获得每张所述热图中不同人体的关键点的坐标，包括：

根据每张所述热图中不同人体的关键点的坐标，将每张所述热图中不同人体的关键点分配给对应的人体，获得每张所述热图中不同人体的关键点的坐标。

3.如权利要求2所述的方法，其特征在于，所述根据每张所述热图中不同人体的关键点的坐标，将每张所述热图中不同人体的关键点分配给对应的人体，获得每张所述热图中不同人体的关键点的坐标，包括：

根据每张所述热图中不同人体的关键点的坐标和每个所述人体的位置，将每张所述热图中不同人体的关键点映射至对应的人体，获得每张所述热图中不同人体的关键点的坐标。

4.如权利要求1所述的方法，其特征在于，所述根据每个所述人体的关键点的坐标，绘制每个所述人体的关节特征图，包括：

根据每个所述人体的关键点的坐标和预设关键点连接规则，绘制每个所述人体的关节特征图。

5.如权利要求1至4任一项所述的方法，其特征在于，所述根据每个所述人体的关键点的坐标，绘制每个所述人体的关节特征图之后，还包括：

通过分类网络对每个所述人体的关节特征图进行分类，获得每个所述人体的姿态类别标签。

6.如权利要求5所述的方法，其特征在于，所通过分类网络对每个所述人体的关节特征图进行分类，获得每个所述人体的姿态类别标签，包括：

使用不同颜色分别标识每个所述人体的关节特征图中的左部分关节和右部分关节；

将标识后的每个所述人体的关节特征图调整至预设大小；

通过调整后的每个所述人体的关节特征图对分类网络进行训练，获得所述分类网络输出的每个所述人体的姿态类别标签。

7.如权利要求6所述的方法，其特征在于，所述分类网络为轻量级分类网络；

所述将标识后的每个所述人体的关节特征图调整至预设大小，包括：

根据所述分类网络的分辨率，将标识后的每个所述人体的关节特征图调整至预设大小；

所述通过调整后的每个所述人体的关节特征图对分类网络进行训练，获得所述分类网络输出的每个所述人体的姿态类别标签，包括：

通过调整后的每个所述人体的关节特征图对分类网络进行训练，获得所述分类网络最后一层的归一化指数损失函数输出的每个所述人体的姿态类别标签。

8.一种人体关键点识别装置，其特征在于，包括：

关键点检测模块，用于通过关键点检测模型检测人体图像中人体的N个关节位置处的关键点，获得所述关键点检测模型输出的N张热图和一张背景图；其中，N为大于或等于2的整数；

关键点分割模块，用于分割出每张所述热图中不同人体的关键点所在区域，并确定每张所述热图中不同人体的关键点所在区域的轮廓；

热图峰值获取模块，用于在所述关键点区域的轮廓内获取热图峰值；

关键点定位模块，用于根据每张所述热图中不同人体的关键点所在区域的轮廓内的热图峰值的坐标，获得每张所述热图中不同人体的关键点的坐标；

特征图绘制模块，用于根据每个所述人体的关键点的坐标，绘制每个所述人体的关节特征图。