[发明专利]关键点的检测方法和装置

权利要求书

1.一种关键点的检测方法，所述方法包括：

获取包含目标的待测图像和训练后的关键点检测模型，将所述待测图像输入所述关键点检测模型，得到包含所述目标的关键点的热力图；

对各个所述热力图进行合并，将合并结果作为多点热力图；

将所述多点热力图和所述待测图像，输入训练后的关键点修正模型，得到包含所述目标的关键点的热力图，将该热力图作为修正后热力图；

确定所述修正后热力图指示的关键点坐标。

2.根据权利要求1所述的方法，其中，所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

3.根据权利要求1或2所述的方法，其中，所述将所述多点热力图和所述待测图像，输入训练后的关键点修正模型，包括：

对所述多点热力图和所述待测图像进行拼接，得到拼接结果；

将所述拼接结果输入训练后的关键点修正模型。

4.根据权利要求1或2所述的方法，其中，所述关键点检测模型输出的热力图指示所述目标的一个关键点；

所述对各个所述热力图进行合并，将合并结果作为多点热力图，包括：

将所述关键点检测模型输出的各个所述热力图进行合并，得到包含该各个所述热力图中关键点的多点热力图，其中，所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量。

5.一种关键点修正模型的训练方法，所述方法包括：

获取包含目标的待测图像和训练后的关键点检测模型，将所述待测图像输入所述关键点检测模型，得到包含所述目标的关键点的热力图；

对各个所述热力图进行合并，将合并结果作为多点热力图；

将所述多点热力图和所述待测图像，输入待训练的关键点修正模型，得到包含所述目标的关键点的热力图，将该热力图作为修正后热力图；

确定所述修正后热力图指示的关键点坐标；

基于所述关键点坐标确定所述关键点修正模型的损失值，通过所述损失值训练所述关键点修正模型，得到训练后的关键点修正模型。

6.根据权利要求5所述的方法，其中，所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。

7.根据权利要求5或6所述的方法，其中，所述将所述多点热力图和所述待测图像，输入训练后的关键点修正模型，包括：

对所述多点热力图和所述待测图像进行拼接，得到拼接结果；

将所述拼接结果输入训练后的关键点修正模型。

8.根据权利要求5或6所述的方法，其中，所述关键点检测模型输出的热力图指示所述目标的一个关键点；

所述对各个所述热力图进行合并，将合并结果作为多点热力图，包括：

将所述关键点检测模型输出的各个所述热力图进行合并，得到包含该各个所述热力图中关键点的多点热力图，其中，所述多点热力图中关键点的数量等于所述关键点检测模型输出的热力图的数量。

9.一种关键点的检测装置，所述装置包括：

获取单元，被配置成获取包含目标的待测图像和训练后的关键点检测模型，将所述待测图像输入所述关键点检测模型，得到包含所述目标的关键点的热力图；

合并单元，被配置成对各个所述热力图进行合并，将合并结果作为多点热力图；

修正单元，被配置成将所述多点热力图和所述待测图像，输入训练后的关键点修正模型，得到包含所述目标的关键点的热力图，将该热力图作为修正后热力图；

确定单元，被配置成确定所述修正后热力图指示的关键点坐标。

10.根据权利要求9所述的装置，其中，所述关键点检测模型和所述关键点修正模型两者的参数数量小于预设数值。