[发明专利]关键点检测方法、装置、电子设备及存储介质

说明书

本公开提供了一种关键点检测方法、装置、电子设备及存储介质，所述方法包括：获取待检测图像；基于所述待检测图像，生成图像特征图和多个关键点热图；所述图像特征图用于表征所述待检测图像中各个目标对象之间的相对位置关系；每个所述关键点热图中包含所述待检测图像的一种类别的关键点，不同类别的关键点对应所述目标对象的不同部位；基于所述图像特征图和多个所述关键点热图，生成关键点图模型；所述关键点图模型中包含所述待检测图像中不同类别的关键点的信息以及连接边的信息，每个连接边为两个不同类别的关键点之间的边；基于所述关键点图模型，确定属于同一目标对象的各个关键点。

技术领域

本公开涉及图像检测技术领域，具体而言，涉及一种关键点检测方法、装置、电子设备及存储介质。

背景技术

人体关键点检测为分析视频中人的行为提供了高层信息支持，是实现视频人体动作识别和人机交互的基础。近年来，基于深度神经网络的人体关键点检测方法，因其输入图像简单易获取、检测效果高效准确，成为了研究热点。

一般的，基于深度神经网络的人体关键点检测方法中，关键点的检测和关键点的聚类是两个独立的步骤，即关键点的聚类一般是不可微分的后处理操作。但是，这种方式下，关键点聚类过程的准确度较低，即可能会产生聚类错误，将不同人的关键点作为同一个人的关键点聚类在一起，从而导致检测结果出现错误。

发明内容

有鉴于此，本公开至少提供一种关键点检测方法、装置、电子设备及存储介质。

第一方面，本公开提供了一种关键点检测方法，包括：

获取待检测图像；

基于所述待检测图像，生成图像特征图和多个关键点热图；所述图像特征图用于表征所述待检测图像中各个目标对象之间的相对位置关系；每个所述关键点热图中包含所述待检测图像的一种类别的关键点，不同类别的关键点对应所述目标对象的不同部位；

基于所述图像特征图和多个所述关键点热图，生成关键点图模型；所述关键点图模型中包含所述待检测图像中不同类别的关键点的信息以及连接边的信息，每个连接边为两个不同类别的关键点之间的边；

基于所述关键点图模型，确定属于同一目标对象的各个关键点。

采用上述方法，可以基于生成的图像特征图和多个关键点热图，生成待检测图像对应的关键点图模型，由于关键点图模型中包括图像特征图和关键点热图中的信息，而图像特征图可以表征出待检测图像中不同目标对象之间的相对位置关系，从而使得基于关键点图模型，可以较准确地对不同目标对象的关键点进行区分，以提高关键点聚类的精准度。

一种可能的实施方式中，基于所述关键点图模型，确定属于同一目标对象的各个关键点，包括：

基于所述关键点图模型中各个关键点的信息以及所述连接边的信息，确定所述关键点图模型中存在连接关系的两个关键点之间的相关度；

基于确定的所述相关度，确定属于同一目标对象的各个关键点。

上述实施方式下，可以基于各个关键点的信息以及连接边的信息，确定关键点图模型中存在连接关系的两个关键点之间的相关度，由于该相关度可以表征对应的两个关键点之间属于同一目标对象的概率，故可以通过确定的相关度，实现将属于同一目标对象的各个关键点聚类在一起，得到每个目标对象对应的各个关键点。

一种可能的实施方式中，基于确定的所述相关度，确定属于同一目标对象的各个关键点，包括：

将对应的所述相关度大于设定阈值的各个关键点作为同一目标对象的关键点。

一种可能的实施方式中，基于所述关键点图模型中各个关键点的信息以及所述连接边的信息，确定所述关键点图模型中存在连接关系的两个关键点之间的相关度，包括：