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

权利要求书

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

获取待检测图像；

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

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

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

2.根据权利要求1所述的方法，其特征在于，基于所述关键点图模型，确定属于同一目标对象的各个关键点，包括：

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

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

3.根据权利要求2所述的方法，其特征在于，基于确定的所述相关度，确定属于同一目标对象的各个关键点，包括：

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

4.根据权利要求2或3所述的方法，其特征在于，基于所述关键点图模型中各个关键点的信息以及所述连接边的信息，确定所述关键点图模型中存在连接关系的两个关键点之间的相关度，包括：

针对每个关键点，基于所述关键点的信息，和所述关键点图模型中与所述关键点存在连接关系的其它关键点的信息，确定所述关键点的融合特征；

基于各个关键点分别对应的融合特征，确定所述关键点图模型中存在连接关系的两个关键点之间的相关度。

5.根据权利要求1～4任一所述的方法，其特征在于，所述关键点的信息包括位置信息、类别信息、以及像素特征信息；

根据以下步骤确定所述关键点图模型中各个关键点的信息：

基于所述关键点热图，确定各个关键点的位置信息；

基于每个所述关键点的位置信息，从所述图像特征图中提取所述关键点的像素特征信息，并基于所述关键点所属关键点热图的类别标签，确定所述关键点对应的类别信息。

6.根据权利要求5所述的方法，其特征在于，根据以下步骤生成所述关键点图模型中的各个连接边：

基于各个关键点对应的所述类别信息，将每个关键点和与所述关键点所属的类别不同的其它关键点连接，形成所述关键点图模型中的连接边。

7.根据权利要求5所述的方法，其特征在于，根据以下步骤生成所述关键点图模型中的各个连接边：

基于各个关键点对应的所述类别信息、和预设的不同类别之间的匹配关系，将每个关键点和与所述关键点所属类别匹配的目标类别对应的关键点连接，形成所述关键点图模型中的连接边。

8.根据权利要求1～7任一所述的方法，其特征在于，在基于所述关键点图模型，确定属于同一目标对象的各个关键点之后，还包括：

基于每个目标对象对应的各个关键点的信息，确定所述目标对象的行为类型。

9.根据权利要求1～7任一所述的方法，其特征在于，在基于所述关键点图模型，确定属于同一目标对象的各个关键点之后，还包括：

基于每个目标对象对应的各个关键点的信息，生成针对所述目标对象的特效信息。