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

权利要求书

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

获取待检测人体图像；

基于预设的关键点检测网络，根据所述待检测人体图像获取人体关键点预测热图；其中，所述关键点检测网络以预设的人体部位的高斯响应热图作为监督信息，所述高斯响应热图根据人体数据集和预设的人体关键点之间的连接关系预先构建获得；所述高斯响应热图包括关键点响应热图和躯干响应热图，所述关键点响应热图以关键点位置为中心进行构建，所述躯干响应热图以关键点之间的躯干连接为中心进行构建；

根据所述人体关键点预测热图对人体关键点的位置进行检测。

2.如权利要求1所述的人体关键点检测方法，其特征在于，所述方法通过以下步骤预先构建所述高斯响应热图：

获取人体数据文件，并对所述人体数据文件进行数据增强处理，相应获得所述人体数据集；其中，所述人体数据文件至少包括人体图像文件、人体检测框的标注文件以及人体关键点位置坐标的标注文件；

根据所述人体数据集和所述人体关键点之间的连接关系构建所述高斯响应热图；其中，所述人体关键点之间的连接关系至少包括人体的头部、脖子、右肩、右肘、右手腕、左肩、左肘、左手腕、胸部、骨盆、左髋、右髋、左膝盖、左脚踝、右膝盖和右脚踝之间的连接关系。

3.如权利要求1或2所述的人体关键点检测方法，其特征在于，所述关键点检测网络包括第一级检测网络、第二级检测网络、第三级检测网络和第四级检测网络；

则，所述基于预设的关键点检测网络，根据所述待检测人体图像获取人体关键点预测热图，具体包括：

根据所述第一级检测网络对所述待检测人体图像进行特征提取，相应获得所述待检测人体图像的第一级特征信息；

根据所述第二级检测网络对所述待检测人体图像进行特征提取，相应获得所述待检测人体图像的第二级特征信息；

根据所述第三级检测网络对所述待检测人体图像进行特征提取，相应获得所述待检测人体图像的第三级特征信息；

根据所述第四级检测网络对所述第一级特征信息、所述第二级特征信息和所述第三级特征信息进行特征融合，相应获得所述人体关键点预测热图；

其中，所述第一级检测网络和所述第二级检测网络以所述躯干响应热图作为监督信息，所述第三级检测网络和所述第四级检测网络以所述关键点响应热图作为监督信息。

4.如权利要求3所述的人体关键点检测方法，其特征在于，所述第一级检测网络、所述第二级检测网络和所述第三级检测网络均由Hourglass网络构成；所述第四级检测网络包括3*3卷积神经网络。

5.如权利要求1所述的人体关键点检测方法，其特征在于，所述根据所述人体关键点预测热图对人体关键点的位置进行检测，具体包括：

根据最大似然估计从所述人体关键点预测热图中获取目标关键点的初始坐标，并获取所述目标关键点的N个相邻点的坐标；其中，N≥2；

采用插值法，根据所述目标关键点的初始坐标以及N个所述相邻点的坐标确定所述目标关键点的位置坐标。

6.如权利要求5所述的人体关键点检测方法，其特征在于，N＝2；则，所述采用插值法，根据所述目标关键点的初始坐标以及N个所述相邻点的坐标确定所述目标关键点的位置坐标，具体包括：

获取所述人体关键点预测热图中所述目标关键点的初始坐标(x₀，y₀)对应的像素值h₀；

获取所述人体关键点预测热图中所述目标关键点的第一相邻点的坐标(x₁，y₁)对应的像素值h₁和第二相邻点的坐标(x₂，y₂)对应的像素值h₂；

采用牛顿插值法，根据公式确定所述目标关键点的位置坐标(x，y)。