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

说明书

本发明实施例公开了一种部位识别方法、装置、终端及存储介质。部位识别方法包括：获取显示有用户的至少一个部位的姿态图像；将所述姿态图像输入至部位识别模型中；获取所述部位识别模型输出的用户的至少一个部位的位置信息；其中，所述部位识别模型包括特征提取子模型和部位检测子模型，所述特征提取子模型包括：多组依次连接的卷积模块和降采样模块；所述部位检测子模型包括：多个依次连接的检测模块和卷积模块；所述检测模块包括：相连接的降采样模块、多组依次连接的卷积模块和升采样模块。本发明实施例提供的部位识别方法运算量小，结构简单，满足直播的实时性要求，能够在终端运行。

技术领域

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

背景技术

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

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

但是，现有的图像识别算法结构复杂、运算量大，需要较大的存储空间和较长的运算时间，导致难以在终端运行，且难以满足直播的实时性要求。

发明内容

本发明实施例提供一种部位识别方法、装置、终端及存储介质，以实现用户部位的快速识别，满足直播的实时性要求，能够在终端运行。

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

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

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

获取所述部位识别模型输出的用户的至少一个部位的位置信息；

其中，所述部位识别模型包括特征提取子模型和部位检测子模型，所述特征提取子模型包括：多组依次连接的卷积模块和降采样模块，用于输入姿态图像，输出所述姿态图像中的特征；

所述部位检测子模型包括：多个依次连接的检测模块和卷积模块，用于输入所述姿态图像中的特征，输出对应部位的位置信息；

所述检测模块包括：相连接的降采样模块、多组依次连接的卷积模块和升采样模块。

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

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

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

第二获取模块，用于获取所述部位识别模型输出的用户的至少一个部位的位置信息；

其中，所述部位识别模型包括特征提取子模型和部位检测子模型，所述特征提取子模型包括：多组依次连接的卷积模块和降采样模块，用于输入姿态图像，输出所述姿态图像中的特征；

所述部位检测子模型包括：多个依次连接的检测模块和卷积模块，用于输入所述姿态图像中的特征，输出对应部位的位置信息；

所述检测模块包括：相连接的降采样模块、多组依次连接的卷积模块和升采样模块。

第三方面，本发明实施例还提供了一种终端，包括：

一个或多个处理器；

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

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