[发明专利]关键点检测方法、装置、设备和计算机可读介质

说明书

本公开的实施例公开了关键点检测方法、装置、电子设备和计算机可读介质。该方法的一具体实施方式包括：对待检测图像进行特征提取，得到图像特征；将图像特征输入预先训练的热力图回归网络，得到关键点热力图；将热力图回归网络的中间层的输出结果输入预先训练的坐标回归网络，得到关键点坐标集合；基于关键点热力图和关键点坐标集合，生成待检测图像对应的关键点位置信息。该实施方式实现了同时满足准确率和关键点之间的关联关系的要求。

技术领域

本公开的实施例涉及计算机技术领域，具体涉及关键点检测方法、装置、设备和计算机可读介质。

背景技术

关键点检测被广泛应用于各种计算机视觉任务。相关的关键点检测技术包括坐标回归和热力图回归。

然而，当采用上述方式进行关键点检测时，经常会存在如下技术问题：

当采用坐标回归的方法时，得到的关键点坐标不够准确。当采用热力图回归的方法时，各个关键点之间的关联关系弱。也就是说，相关的检测方式无法同时满足准确率和关键点之间的关联关系的要求。

发明内容

本公开的内容部分用于以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。本公开的内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。本公开的一些实施例提出了关键点检测方法、装置、电子设备和计算机可读介质，来解决以上背景技术部分提到的技术问题中的一项或多项。

第一方面，本公开的一些实施例提供了一种关键点检测方法，该方法包括：对待检测图像进行特征提取，得到图像特征；将图像特征输入预先训练的热力图回归网络，得到关键点热力图；将热力图回归网络的中间层的输出结果输入预先训练的坐标回归网络，得到关键点坐标集合；基于关键点热力图和关键点坐标集合，生成待检测图像对应的关键点位置信息。

第二方面，本公开的一些实施例提供了一种关键点检测装置，装置包括：提取单元，被配置成对待检测图像进行特征提取，得到图像特征；热力图生成单元，被配置成将图像特征输入预先训练的热力图回归网络，得到关键点热力图；坐标生成单元，被配置成将热力图回归网络的中间层的输出结果输入预先训练的坐标回归网络，得到关键点坐标集合；位置信息生成单元，被配置成基于关键点热力图和关键点坐标集合，生成待检测图像对应的关键点位置信息。

第三方面，本公开的一些实施例提供了一种电子设备，包括：一个或多个处理器；存储装置，其上存储有一个或多个程序，当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现上述第一方面任一实现方式所描述的方法。

第四方面，本公开的一些实施例提供了一种计算机可读介质，其上存储有计算机程序，其中，程序被处理器执行时实现上述第一方面任一实现方式所描述的方法。

本公开的上述各个实施例具有如下有益效果：通过本公开的一些实施例的关键点检测方法，结合了热力图回归网络和坐标回归网络，从而将两种方式的优势进行结合。具体来说，热力图回归网络预测的关键点坐标准确，坐标回归网络预测的关键点之间的关联关系强，因此，可以同时满足准确率和关键点之间的关联关系的要求。

附图说明

结合附图并参考以下具体实施方式，本公开各实施例的上述和其他特征、优点及方面将变得更加明显。贯穿附图中，相同或相似的附图标记表示相同或相似的元素。应当理解附图是示意性的，元件和元素不一定按照比例绘制。

图1是本公开的一些实施例的关键点检测方法的一个应用场景的示意图；

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

图3示例性示出了利用热力图回归网络得到的手部关键点检测结果；

图4示例性示出了利用坐标回归网络得到的手部关键点检测结果；