[发明专利]部位识别方法、装置、设备及存储介质

说明书

本发明实施例公开了一种部位识别方法、装置、设备及存储介质。方法包括：获取显示有至少一个部位的姿态图像；将姿态图像输入至部位识别模型中；获取部位识别模型输出的识别结果；部位识别模型包括：至少两个并行排布且内部斜向连接的部位识别子模型，部位识别子模型用于识别对应的部位，包括顺序连接的至少两个特征提取单元，不同顺次的特征提取单元对应不同的计算级数。本实施例提供的方法提高对姿态图像的处理能力和处理速度，以及提高了整体网络的表达能力。

技术领域

本发明实施例涉及计算机视觉技术，尤其涉及一种部位识别方法、装置、设备及存储介质。

背景技术

随着直播技术的发展，越来越多的用户通过直播视频获取信息、参与活动，例如通过直播视频进行姿态矫正或者动作引导。

在进行姿态矫正或者动作引导的过程中较重要的环节是识别用户的部位。用户的各部位包括各肢体、关节以及眼耳口鼻等器官。随着计算机视觉技术的发展，可以通过图像识别算法识别图像中各部位的位置、尺寸等信息，再辅以主播的讲解，为用户提供姿态矫正或者动作引导建议。

但是，现有的图像识别算法的识别速率较慢、精度不高，难以满足直播的实时性要求和准确性要求，进而导致直播视频的时延和矫正不及时、不准确的后果。

发明内容

本发明实施例提供一种部位识别方法、装置、设备及存储介质，以提高对姿态图像的识别精度和速率。

第一方面，本发明实施例提供了一种部位识别方法，包括：

获取显示有至少一个部位的姿态图像；

将所述姿态图像输入至部位识别模型中；

获取所述部位识别模型输出的识别结果；

其中，所述部位识别模型包括：至少两个并行排布且内部斜向连接的部位识别子模型，所述部位识别子模型用于识别对应的部位，包括顺序连接的至少两个特征提取单元，不同顺次的特征提取单元对应不同的计算级数；

至少一个所述特征提取单元，用于对至少两个所述部位识别子模型中的至少两个第一特征提取单元的输出特征进行特征提取，和/或，将特征提取后的结果输入至至少两个所述部位识别子模型中的至少两个第二特征提取单元中，其中，所述第一特征提取单元的计算级数位于所述特征提取单元的计算级数之前，所述第二特征提取单元的计算级数位于所述特征提取单元的计算级数之后。

第二方面，本发明实施例还提供了一种部位识别装置，该装置包括：

第一获取模块，用于获取显示有至少一个部位的姿态图像；

输入模块，用于将所述姿态图像输入至部位识别模型中；

第二获取模块，用于获取所述部位识别模型输出的识别结果；

其中，所述部位识别模型包括：至少两个并行排布且内部斜向连接的部位识别子模型，所述部位识别子模型用于识别对应的部位，包括顺序连接至少两个特征提取单元，不同顺次的特征提取单元对应不同的计算级数；

至少一个所述特征提取单元，用于对至少两个所述部位识别子模型中的至少两个第一特征提取单元的输出特征进行特征提取，和/或，将特征提取后的结果输入至至少两个所述部位识别子模型中的至少两个第二特征提取单元中，其中，所述第一特征提取单元的计算级数位于所述特征提取单元的计算级数之前，所述第二特征提取单元的计算级数位于所述特征提取单元的计算级数之后。

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

一个或多个处理器；

存储器，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现任一实施例所述的部位识别方法。