[发明专利]一种人体坐姿识别方法、装置、电子设备和存储介质

权利要求书

1.一种人体坐姿识别方法，包括：

采集图片；

获取所述图片对应的多张预测图，并获取每张预测图中的至少一个人体预测框；

从维度最低的所述预测图中获取每个所述人体预测框对应的人体局部特征图；

生成每个所述人体局部特征图对应的人体关键点热力图和注意力权重，并根据所述人体局部特征图与对应的所述注意力权重得到对应的加权人体局部特征图；

对所有的加权人体局部特征图进行识别得到所述图片的人体坐姿分类结果。

2.根据权利要求1所述的方法，其中，所述获取所述图片对应的多张预测图，包括：

通过主干网络对采集的所述图片进行特征提取，得到多张网格图；

对所述网格图进行卷积处理和上采样处理，得到每张网格图对应的预测图。

3.根据权利要求1所述的方法，其中，所述主干网络为轻量级网络。

4.根据权利要求1所述的方法，其中，所述从维度最低的所述预测图中获取每个所述人体预测框对应的人体局部特征图，包括：

根据所述人体预测框的尺寸，从所述维度最低的所述预测图中裁剪出所述人体预测框对应的人体局部特征图。

5.根据权利要求1或4所述的方法，其中，所述获取每个所述人体预测框对应的人体局部特征图之后，该方法还包括：

将每个所述人体局部特征图的尺寸转换为设定尺寸。

6.根据权利要求5所述的方法，其中，所述生成每个所述人体局部特征图对应的人体关键点热力图和注意力权重，包括：

获取所述人体局部特征图中人体关键点的位置信息；

根据所述人体关键点的位置信息，生成所述人体关键点热力图和所述注意力权重。

7.根据权利要求1所述的方法，其中，所述根据所述人体局部特征图与对应的所述注意力权重得到对应的加权人体局部特征图，包括：

将所述人体局部特征图与对应的所述注意力权重进行加权相乘后得到所述加权人体局部特征图。

8.一种人体坐姿识别的装置，包括：

采集模块，用于采集图片；

第一图片处理模块，用于获取所述图片对应的多张预测图，并获取每张预测图中的至少一个人体预测框；

第二图片处理模块，用于从维度最低的所述预测图中获取每个所述人体预测框对应的人体局部特征图；

第三图片处理模块，生成每个所述人体局部特征图对应的人体关键点热力图和注意力权重，并根据所述人体局部特征图与对应的所述注意力权重得到对应的加权人体局部特征图；

输出模块，对所有的加权人体局部特征图进行识别得到所述图片的人体坐姿分类结果。

9.一种电子设备，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-7中任一项所述的方法。

10.一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行根据权利要求1-7中任一项所述的方法。

11.一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现根据权利要求1-7中任一项所述的方法。