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

权利要求书

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

获取目标视频中第一视频帧的多点热力图，其中，所述多点热力图包含所述第一视频帧中目标的各个关键点；

将所述多点热力图和所述目标视频中的第二视频帧，输入轻量关键点检测模型，得到包含所述第二视频帧中目标的关键点的单点热力图，其中，所述轻量关键点检测模型的参数数量小于指定数值，所述第一视频帧与所述第二视频帧相差的视频帧数量不大于预设差值；

根据各个所述单点热力图，确定所述第二视频帧中目标的关键点坐标值，包括：对各个所述单点热力图进行合并，得到对应的多点热力图；将该多点热力图指示的关键点坐标值，作为所述第二视频帧中目标的关键点坐标值。

2.根据权利要求1所述的方法，其中，所述第一视频帧为所述视频中检测的起始视频帧；

所述获取目标视频中第一视频帧的多点热力图，包括：

将所述起始视频帧，输入精确关键点检测模型，得到包含所述起始视频帧中目标的关键点的单点热力图，其中，所述精确关键点检测模型的参数数量大于所述轻量关键点检测模型的参数数量；

对各个该单点热力图进行合并，根据合并结果生成所述多点热力图。

3.根据权利要求1所述的方法，其中，所述将所述多点热力图和所述目标视频中的第二视频帧，输入轻量关键点检测模型，包括：

对所述多点热力图和所述第二视频帧进行拼接，得到拼接结果；

将所述拼接结果输入轻量关键点检测模型。

4.根据权利要求1或3所述的方法，其中，所述轻量关键点检测模型的训练步骤，包括：

获取训练样本，其中，所述训练样本中包括样本图像，和基于所述样本图像得到的样本多点热力图；

将所述样本多点热力图和所述样本图像，输入待训练的轻量关键点检测模型，得到所述样本图像中目标的关键点的单点热力图；

基于该单点热力图指示的关键点坐标值，确定所述待训练的轻量关键点检测模型的损失值，通过所述损失值训练所述待训练的轻量关键点检测模型。

5.根据权利要求2所述的方法，其中，所述轻量关键点检测模型的训练步骤，包括：

获取训练样本，其中，所述训练样本中包括样本图像，和基于所述样本图像得到的样本多点热力图；

将所述样本多点热力图和所述样本图像，输入待训练的轻量关键点检测模型，得到所述样本图像中目标的关键点的单点热力图；

基于该单点热力图指示的关键点坐标值，确定所述待训练的轻量关键点检测模型的损失值，通过所述损失值训练所述待训练的轻量关键点检测模型。

6.根据权利要求5所述的方法，其中，所述样本多点热力图的生成步骤，包括：

将所述样本图像输入所述精确关键点检测模型，得到包含所述样本图像中目标的关键点的单点热力图；

根据合并各个该单点热力图所得到的合并结果，生成所述样本多点热力图。

7.根据权利要求4所述的方法，其中，所述样本图像为样本视频的视频帧；

所述样本多点热力图的生成步骤，包括：

获取所述样本视频中各个视频帧的热力图真值，其中，所述热力图真值为单点热力图真值或多点热力图真值；

对所述各个视频帧的热力图真值进行随机扰动，根据扰动后关键点热力图，确定所述各个视频帧分别对应的多点热力图，并将该多点热力图作为所述样本多点热力图。

8.根据权利要求5所述的方法，其中，所述样本图像为样本视频的视频帧；

所述样本多点热力图的生成步骤，包括：

获取所述样本视频中各个视频帧的热力图真值，其中，所述热力图真值为单点热力图真值或多点热力图真值；

对所述各个视频帧的热力图真值进行随机扰动，根据扰动后关键点热力图，确定所述各个视频帧分别对应的多点热力图，并将该多点热力图作为所述样本多点热力图。