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

权利要求书

1.一种部位识别方法，其特征在于，包括：

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

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，至少一组相同计算级数的特征提取单元不连接。

3.根据权利要求1所述的方法，其特征在于，每个部位识别子模型包括顺序连接的1个起始特征提取单元、N个中间特征提取单元以及1个结尾特征提取单元；

起始特征提取单元，用于对所述姿态图像进行特征提取，并将特征提取后的结果输入至各所述部位识别子模型中第1个中间特征提取单元中；

第M个中间特征提取单元，用于对各所述部位识别子模型中第M-1个特征提取单元的输出特征进行特征提取，并将特征提取后的结果输入至各所述部位识别子模型中第M+1个特征提取单元中；其中，各所述部位识别子模型中第M-1个特征提取单元的计算级数低于第M个中间特征提取单元，各所述部位识别子模型中第M+1个特征提取单元的计算级数高于第M个中间特征提取单元；

结尾特征提取单元，用于对各所述部位识别子模型中第N个中间特征提取单元的输出特征进行特征提取，并输出识别结果；

其中，M和N是自然数，1≤M≤N。

4.根据权利要求1所述的方法，其特征在于，

所述部位识别模型中部位识别子模型的个数与待识别的部位的个数相等；

所述部位识别子模型中特征提取单元的个数为Q个，其中，Q是自然数，2≤Q≤8。

5.根据权利要求1-4中任一项所述的方法，其特征在于，所述特征提取单元包括至少一个沙漏单元；

所述沙漏单元包括多个Residual模块。

6.根据权利要求1所述的方法，其特征在于，所述获取所述部位识别模型输出的识别结果，包括：

获取各所述部位识别子模型输出的对应部位的热力图。

7.根据权利要求6所述的方法，其特征在于，在获取各所述部位识别子模型输出的对应部位的热力图之后，还包括：

从对应各部位的热力图中，分别确定各部位的位置信息。

8.一种部位识别装置，其特征在于，包括：

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

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

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

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

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