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

说明书

本发明适用于人工智能技术领域，提供一种人体关键点识别方法、装置、终端设备及存储介质，通过关键点检测模型检测人体图像中人体的N个关节位置处的关键点，获得所述关键点检测模型输出的N张热图和一张背景图；分割出每张所述热图中不同人体的关键点所在区域，并确定每张所述热图中不同人体的关键点所在区域的轮廓；在所述关键点区域的轮廓内获取热图峰值；根据每张所述热图中不同人体的关键点所在区域的轮廓内的热图峰值的坐标，获得每张所述热图中不同人体的关键点的坐标；根据每个所述人体的关键点的坐标，绘制每个所述人体的关节特征图，可以有效提高人体的关键点的识别精度。

技术领域

本发明属于人工智能(Artificial Intelligence，AI)技术领域，尤其涉及一种人体关键点识别方法、装置、终端设备及存储介质。

背景技术

随着人工智能技术的快速发展，基于神经网络的目标检测及分类技术已经较为成熟，在工业领域应用广泛，例如，利用图像对人体的姿态进行识别。人体的姿态可以分为静态和动态，静态动作(例如，站立、静坐、举手等)依靠单帧图像即可判别，常用的静态动作识别方法是基于目标检测的方法；动态动作(例如，行走、跳跃、跑步等)依靠多帧连续的图像序列进行判别，常用的动态动作识别方法是基于双流网络或3D卷积等的方法。人体关键点识别是静态动作识别的实现基础，保证人体关键点的识别精度，是解决现有的静态动作识别方法，存在的训练所需的数据量大，涵盖图中的服饰、背景等，导致训练速度慢且可解释性较弱等问题的关键。

发明内容

有鉴于此，本发明实施例提供了一种人体关键点识别方法、装置、终端设备及存储介质，能够有效提高人体关键点的识别精度。

本发明实施例的第一方面提供了一种人体关键点识别方法，包括：

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

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

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

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

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

本发明实施例的第二方面提供了一种人体关键点识别装置，包括：

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

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

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

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

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

本发明实施例的第三方面提供了一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如本发明实施例的第一方面所述人体关键点识别方法的步骤。

本发明实施例的第四方面提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现如本发明实施例的第一方面所述人体关键点识别方法的步骤。