[发明专利]一种人体骨骼点检测方法、装置、电子设备和存储介质

说明书

本发明涉及图像数据处理技术，公开了一种人体骨骼点检测方法、装置、电子设备和存储介质，包括：采集第一原始图像，识别第一原始图像中至少一个人物的第一外接矩形框的第一位置信息；根据第一位置信息，确定第一原始图像中人物的第二外接矩形框的第二位置信息；基于第二外接矩形框的第二位置信息对第一原始图像进行裁剪，得到人物对应的第一裁剪图像；将第一裁剪图像输入至预先训练的卷积神经网络模型，得到人物的人体骨骼点位置信息集合，并根据人体骨骼点位置信息集合更新人物的第一外接矩形框的第一位置信息。提高了人体骨骼点检测准确率和检测效率的同时，也缩短了检测时间。

技术领域

本发明涉及图像数据处理技术领域，尤其涉及一种人体骨骼点检测方法、装置、电子设备和存储介质。

背景技术

现有方案中，从图像中检测人体骨骼点方法大概分两类：第一类使用自上而下方式，先检测图像中每个人的位置，用矩形框表示，然后将每个矩形框中的图像裁切出来，输入到模型中得到图像中所有的人体骨骼点；第二类使用自下而上的方式，先检测图片中所有骨骼点的位置，然后把骨骼点连接起来，组成每个人的人体骨骼点。

然而，发明人在实现本发明的过程中发现，第一类方案首先运行一个检测人体的模型来计算每个人的外接矩形框，然后再运行骨骼点检测模型来检测骨骼点位置，如果一张图像上有n个人，则骨骼点检测模型需要运行n次，所以检测一张图像，计算时间长且效率低；第二类方案通过骨骼点识别人体，准确率较低，且当图像中人数比较少时，运算会带进大量背景，大量的算力被浪费，效率过低。

发明内容

为了解决上述技术问题或者至少部分地解决上述技术问题，本发明实施例提供了一种人体骨骼点检测方法、装置、电子设备和存储介质，提高了识别准确率以及人体骨骼点检测效率，也缩短了检测时间。

本发明实施例提供了一种人体骨骼点检测方法，包括：

根据第一外接矩形框的第一位置信息，确定第一原始图像中人物的第二外接矩形框的第二位置信息；其中，第二外接矩形框的高、宽分别大于其对应的第一外接矩形框的高、宽；基于第二外接矩形框的第二位置信息对第一原始图像进行裁剪，得到人物对应的第一裁剪图像；将第一裁剪图像输入至预先训练的卷积神经网络模型，得到人物的人体骨骼点位置信息集合，并根据人体骨骼点位置信息集合更新人物的第一外接矩形框的第一位置信息；其中，第一位置信息用于后续采集图像中人物的第一外接矩形框的定位。

本发明实施例还提供了一种人体骨骼点检测装置，包括：

识别模块，用于采集第一原始图像，识别第一原始图像中至少一个人物的第一外接矩形框的第一位置信息；确定模块，用于根据第一外接矩形框的第一位置信息，确定第一原始图像中人物的第二外接矩形框的第二位置信息；其中，第二外接矩形框的高、宽分别大于其对应的第一外接矩形框的高、宽；裁剪模块，用于基于第二外接矩形框的第二位置信息对第一原始图像进行裁剪，得到人物对应的第一裁剪图像；生成模块，用于将第一裁剪图像输入至预先训练的卷积神经网络模型，得到人物的人体骨骼点位置信息集合，并根据人体骨骼点位置信息集合更新人物的第一外接矩形框的第一位置信息；其中，第一位置信息用于后续采集图像中人物的第一外接矩形框的定位。

本发明实施例还提供了一种电子设备，包括：

一个或多个处理器；存储装置，用于存储一个或多个程序；当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现如上所述的人体骨骼点检测方法。

本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如上的人体骨骼点检测方法。

本发明实施例还提供了一种计算机程序产品，该计算机程序产品包括计算机程序或指令，该计算机程序或指令被处理器执行时实现如上所述的人体骨骼点检测方法。