[发明专利]人体姿态识别模型的训练方法、装置及存储介质

权利要求书

1.一种人体姿态识别模型的训练方法，其特征在于，所述方法包括：

将第一图像输入至人体姿态识别模型，得到第一图像对应的第一特征图；所述第一图像携带有人体姿态关键点；所述第一特征图的第二分辨率低于所述第一图像的第一分辨率；

对所述第一特征图中的每个像素进行至少两次卷积，生成所述第一特征图对应的第一热图；所述第一热图的分辨率大于第一特征图的第二分辨率；所述至少两次卷积使用的是1×1的卷积核；

基于所述第一热图中每个像素的第一参数和第二参数确定损失值；所述第一参数表征对应像素在所述第一热图的高斯置信值；所述第二参数表征对应像素在第二热图的高斯置信值；所述第二热图表征基于所述第一图像中携带的关键点确定出的热图；

根据确定出的损失值更新人体姿态识别模型的权重参数；

所述对所述第一特征图中的每个像素进行至少两次卷积，生成所述第一特征图对应的第一热图，包括：

对所述第一特征图中的每个第一像素进行至少两次卷积，生成对应的至少两个第二像素；每个第二像素由第一像素经过一次卷积得到；

按照所述第一特征图中的第一像素的排列顺序排列每个第一像素对应的第二像素，得到第二图像；

基于所述第二图像生成所述第一热图。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

基于所述第二图像的第三分辨率与所述第二分辨率的倍数确定所述至少两次卷积的卷积次数。

3.根据权利要求1所述的方法，其特征在于，所述将第一图像输入至人体姿态识别模型之前，所述方法还包括：

对设定的样本图库中的第三图像进行人体目标检测，并基于人体目标检测过程中定位的目标矩形框对所述第三图像进行裁剪，得到所述第一图像。

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

计算最近两次确定出的损失值的差值；

在所述差值处于设定阈值范围的情况下，停止对所述人体姿态识别模型的迭代训练。

5.一种人体姿态识别方法，其特征在于，所述方法包括：

将包含人体姿态的第四图像输入至人体姿态识别模型，输出所述第四图像对应的第二特征图；

对所述第二特征图中的每个像素进行至少两次卷积，生成所述第二特征图对应的第三热图；所述至少两次卷积使用的是1×1的卷积核；

基于所述第三热图进行人体姿态识别；其中，

所述人体姿态识别模型采用如权利要求1至4任一项所述的人体姿态识别模型的训练方法训练得到；

所述对所述第二特征图中的每个像素进行至少两次卷积，生成所述第二特征图对应的第三热图，包括：

对所述第二特征图中的每个第三像素进行至少两次卷积，生成对应的至少两个第四像素；每个第四像素由第三像素经过一次卷积得到；

按照所述第二特征图中的第三像素的排列顺序排列每个第三像素对应的第四像素，得到第五图像；

基于所述第五图像生成所述第三热图。