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

权利要求书

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

通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；

根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，其中，所述姿态关键点模板表征所述姿态关键点模板中多个关键点的相对位置关系；

对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果。

2.根据权利要求1所述的方法，其特征在于，根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，包括：

根据多个姿态关键点模板，对所述特征图进行关键点检测，得到多个候选关键点偏移量和各候选关键点偏移量对应的置信度，其中，所述候选关键点偏移量为所述特征图中特征点与每个姿态关键点模板中关键点的偏移量；

根据所述多个候选关键点偏移量和所述置信度，确定所述目标对象对应的至少一组关键点。

3.根据权利要求2所述的方法，其特征在于，对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果，包括：

根据所述至少一组关键点和每组组关键点对应的置信度，确定所述目标对象的关键点检测结果。

4.根据权利要求3所述的方法，其特征在于，所述根据所述至少一组关键点和每组关键点对应的置信度，确定所述目标对象的关键点检测结果，包括：

分别确定每组关键点对应的目标框；

根据每组关键点对应的置信度，对所述至少一组关键点对应的目标框进行非极大值抑制处理，得到所述目标对象的关键点检测结果。

5.根据权利要求4所述的方法，其特征在于，所述分别确定每组关键点对应的目标框，包括：

分别将每组关键点对应的最小外接矩形确定为每组关键点对应的目标框。

6.根据权利要求2-5任一项所述的方法，其特征在于，根据所述多个候选关键点偏移量和所述置信度，确定所述目标对象对应的至少一组关键点，包括：

从所述多个候选关键点偏移量中筛选出置信度大于或等于置信度阈值的关键点偏移量的组合，将筛选出的关键点偏移量的组合确定为至少一组关键点偏移量，每组关键点偏移量表征所述特征图中该组关键点与每个姿态关键点模板中关键点的偏移量；

根据所述至少一组关键点偏移量和每组关键点偏移量对应的姿态关键点模板，确定所述目标对象对应的至少一组关键点。

7.根据权利要求6所述的方法，其特征在于，所述对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果，包括：

根据所述至少一组关键点和每组关键点对应的置信度，对所述至少一组关键点进行非极大值抑制处理，得到所述目标对象的关键点检测结果；

所述方法还包括：

根据所述关键点检测结果，确定目标框，所述目标框表征目标对象所在的位置。

8.根据权利要求2-7任一项所述的方法，其特征在于，根据多个姿态关键点模板，对所述特征图进行关键点检测，得到多个候选关键点偏移量和各候选关键点偏移量对应的置信度，包括：

分别基于所述特征图中的每个特征点，将多个所述姿态关键点模板与所述特征图进行匹配，确定所述特征点与每个姿态关键点模板中关键点的偏移量和所述特征点的置信度，得到至少一组关键点偏移量和与每组关键点偏移量对应的置信度。

9.根据权利要求1-8任一项所述的方法，其特征在于，根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，包括：

根据多个姿态关键点模板，通过关键点检测模型对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点偏移量和与每组关键点偏移量对应的置信度。

10.根据权利要求8所述的方法，其特征在于，还包括：基于所述姿态关键点模板和样本图像，对初始关键点检测模型进行训练，得到所述关键点检测模型。