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

说明书

本发明公开了一种人体关键点检测方法，包括：获取待检测人体图像；基于预设的关键点检测网络，根据所述待检测人体图像获取人体关键点预测热图；其中，所述关键点检测网络以预设的人体部位的高斯响应热图作为监督信息，所述高斯响应热图根据人体数据集和预设的人体关键点之间的连接关系预先构建获得；根据所述人体关键点预测热图对人体关键点的位置进行检测。相应的，本发明还公开了一种人体关键点检测装置、计算机可读存储介质及终端设备。采用本发明的技术方案能够结合人体结构信息之间的约束进行人体关键点检测，从而减小量化误差，提高检测结果的准确率。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种人体关键点检测方法、装置、计算机可读存储介质及终端设备。

背景技术

人体关键点检测旨在检测图像上人体的关节、五官等的具体位置，以描述人体的姿态，在自然图像中，拍摄场景的复杂性和人体姿势多样性给人体姿态估计任务带来了巨大挑战。作为计算机视觉技术领域最具挑战性的课题之一，多样化的人体姿态估计数据集的出现以及深度学习技术的发展，使得人体关键点检测的准确率也在逐渐提升，常用的具有代表性的人体关键点检测网络有Hourglass、CPN、MSPN和HRNet等，这些基于置信度热图回归的人体关键点检测网络，大都采用了多级特征融合模块来捕捉丰富的多尺度信息，但牺牲了网络的复杂度，忽略了人体结构信息的空间约束，导致获取关键点的具体位置时，会有一定的量化误差，使得人体关键点检测技术难以满足实际应用中高准确率的需求。

发明内容

本发明实施例所要解决的技术问题在于，提供一种人体关键点检测方法、装置、计算机可读存储介质及终端设备，能够结合人体结构信息之间的约束进行人体关键点检测，从而减小量化误差，提高检测结果的准确率。

为了解决上述技术问题，本发明实施例提供了一种人体关键点检测方法，包括：

获取待检测人体图像；

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

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

进一步地，所述方法通过以下步骤预先构建所述高斯响应热图：

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

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

进一步地，所述关键点检测网络包括第一级检测网络、第二级检测网络、第三级检测网络和第四级检测网络；

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

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

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

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