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

说明书

本公开提供了关键点的检测方法和装置，涉及深度学习和视频等人工智能技术领域。具体实施方式包括：获取目标视频中第一视频帧的多点热力图，其中，多点热力图包含第一视频帧中目标的各个关键点；将多点热力图和目标视频中的第二视频帧，输入轻量关键点检测模型，得到包含第二视频帧中目标的关键点的单点热力图，其中，轻量关键点检测模型的参数数量小于指定数值，第一视频帧与第二视频帧相差的视频帧数量不大于预设差值；根据各个单点热力图，确定第二视频帧中目标的关键点坐标值。本公开可以利用视频中第一视频帧的多点热力图，和相近的第二视频帧，一起输入轻量模型，来检测第二视频帧，从而实现快速、准确的检测过程。

技术领域

本公开涉及计算机技术领域，具体涉及深度学习和视频等人工智能技术领域，尤其涉及关键点的检测方法和装置。

背景技术

姿态估计通常利用物体的几何关系或者物体的关键点来估计。具体地，可以对图像或视频每一帧中的关键点进行检测，利用检测结果实现姿态估计。

相关技术中，比如人体的姿态估计，往往检测的关键点是人体的关节点坐标位置。具体地，可以先使用检测器检测出指示人所在位置的目标框，之后，再从目标框中检测出人体的关节点的坐标位置。

发明内容

提供了一种关键点的检测方法、装置、电子设备以及存储介质。

根据第一方面，提供了一种关键点的检测方法，包括：获取目标视频中第一视频帧的多点热力图，其中，多点热力图包含第一视频帧中目标的各个关键点；将多点热力图和目标视频中的第二视频帧，输入轻量关键点检测模型，得到包含第二视频帧中目标的关键点的单点热力图，其中，轻量关键点检测模型的参数数量小于指定数值，第一视频帧与第二视频帧相差的视频帧数量不大于预设差值；根据各个单点热力图，确定第二视频帧中目标的关键点坐标值。

根据第二方面，提供了一种关键点的检测装置，包括：获取目标视频中第一视频帧的多点热力图，其中，多点热力图包含第一视频帧中目标的各个关键点；将多点热力图和目标视频中的第二视频帧，输入轻量关键点检测模型，得到包含第二视频帧中目标的关键点的单点热力图，其中，轻量关键点检测模型的参数数量小于指定数值，第一视频帧与第二视频帧相差的视频帧数量不大于预设差值；根据各个单点热力图，确定第二视频帧中目标的关键点坐标值。

根据第三方面，提供了一种电子设备，包括：至少一个处理器；以及与至少一个处理器通信连接的存储器；其中，存储器存储有可被至少一个处理器执行的指令，指令被至少一个处理器执行，以使至少一个处理器能够执行关键点的检测方法中任一实施例的方法。

根据第四方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，计算机指令用于使计算机执行根据关键点的检测方法中任一实施例的方法。

根据第五方面，提供了一种计算机程序产品，包括计算机程序，计算机程序在被处理器执行时实现根据关键点的检测方法中任一实施例的方法。

根据本公开的方案，可以利用视频中第一视频帧的多点热力图，和相近的第二视频帧，一起输入轻量模型，来检测第二视频帧，从而实现快速、准确的检测过程。

附图说明

通过阅读参照以下附图所作的对非限制性实施例所作的详细描述，本公开的其它特征、目的和优点将会变得更明显：

图1是本公开一些实施例可以应用于其中的示例性系统架构图；

图2a是根据本公开的关键点的检测方法的一个实施例的流程图；

图2b是根据本公开的关键点的检测方法的又一个实施例的流程图；

图3是根据本公开的关键点的检测方法的一个应用场景的示意图；

图4是根据本公开的关键点的检测方法的又一个实施例的流程图；

图5是根据本公开的关键点的检测装置的一个实施例的结构示意图；