[发明专利]人体关键点检测方法和装置、电子设备、计算机存储介质

说明书

本发明实施例公开了一种人体关键点检测方法和神经网络训练方法、装置、电子设备、计算机存储介质，该人体关键点检测方法包括：获取待处理图像；利用预先训练的神经网络检测所述待处理图像中的人体关键点，其中，根据每个人体关键点及其相邻关键点的损失得到所述预先训练的神经网络的损失函数，并根据所述损失函数训练神经网络，得到所述预先训练的神经网络，所述人体关键点的相邻关键点由各人体关键点之间的连接关系确定。

技术领域

本发明涉及计算机视觉领域，尤其涉及一种人体关键点检测方法和神经网络训练方法、装置、电子设备、计算机存储介质，可以应用于智能视频分析或安防监控。

背景技术

人体关键点检测技术是人体视频数据自动处理、人体行为分析和人机交互的基础，可以为视频结构化提供重要的技术支撑；人体关键点检测的主要原理是：从人体上检测出各个人体关键点(可以是关节或者五官位置)，从而通过这些关键点的位置信息来描述人体的姿态。

在相关技术中，如何提高人体关键点检测的准确率，是亟待解决的问题和重要的研究方向。

发明内容

本发明实施例期望提供一种人体关键点检测和神经网络训练的技术方案。

本发明实施例提供了一种人体关键点检测方法，所述方法包括：

获取待处理图像；

利用预先训练的神经网络检测所述待处理图像中的人体关键点，其中，根据每个人体关键点及其相邻关键点的损失得到所述预先训练的神经网络的损失函数，并根据所述损失函数训练神经网络，得到所述预先训练的神经网络，所述人体关键点的相邻关键点由各人体关键点之间的连接关系确定。

可选地，所述方法还包括：

根据各人体关键点之间的连接关系的先验知识，预先定义各人体关键点之间的连接关系。

可选地，所述根据每个人体关键点及其相邻关键点的损失得到所述预先训练的神经网络的损失函数，包括：

根据所有人体关键点的平均损失得到所述损失函数，其中，所述关键点的损失由所述关键点及其相邻关键点的损失得到。

可选地，所述关键点的损失由所述关键点及其相邻关键点的损失得到，包括：

将所述关键点的损失及其相邻关键点的损失叠加，得到所述关键点的损失。

可选地，所述损失函数包括：对数损失函数或均方差损失函数。

本发明实施例还提供了一种神经网络训练方法，所述方法还包括：

获取包含人体的样本图像；

根据所述样本图像中每个人体关键点及其相邻关键点的损失获取损失函数，并根据所述损失函数训练用于人体关键点检测的神经网络，其中，每个人体关键点的相邻关键点根据各人体关键点之间的连接关系确定。

可选地，所述方法还包括：

根据各人体关键点之间的连接关系的先验知识，预先定义各人体关键点之间的连接关系。

可选地，所述根据所述样本图像中每个人体关键点及其相邻关键点的损失获取损失函数，包括：

根据所述样本图像中所有人体关键点的平均损失得到所述损失函数，其中，所述关键点的损失由所述关键点及其相邻关键点的损失得到。

可选地，所述关键点的损失由所述关键点及其相邻关键点的损失得到，包括：

将所述关键点的损失及其相邻关键点的损失叠加，得到所述关键点的损失。

可选地，所述损失函数包括：对数损失函数或均方差损失函数。