基于结构指导深度学习的人体姿态估计方法和系统与流程

文档序号:11143412阅读:来源:国知局

技术特征:

1.一种基于结构指导深度学习的人体姿态估计方法,其特征在于,包括如下步骤:

步骤1:输入数据预处理;

步骤2:结构指导卷积神经网络预测;

步骤3:融合卷积神经网络预测;

步骤4:输出处理。

2.根据权利要求1所述的一种基于结构指导深度学习的人体姿态估计方法,其特征在于,所述步骤2进一步具体为:将预处理后的图片通过结构指导的卷积神经网络来进行学习和预测,这里结构指导指的是在一般的卷积神经网络中融入人体骨架的结构信息,来指导神经网络的学习,一般的人体姿态估计的神经网络的回归目标函数为:

表示一张由真实标记关节点为中心产生的真实标记的热点图,其分布为高斯分布yk代表第k真实标记的关节点,(xk,yk)是yk在图像中坐标点的值,(i,j)是真实标记的热点图中的每个像素点在图中的位置,σ1是高斯分布的标准差;结构指导的卷积神经网络加入了骨架结构信息,这部分的目标函数为:

表示由两个相邻的真实标记关节点的连线为中心产生的高斯分布的真实标记热点图,pl表示相邻关节点之间的连线,l是这些连线相应的序号,这些连线对应于不同的人体模型,可以采用树形结构的人体模型,相应的连线即位树形结构人体模型的边;dist((i,j),pl)表示真实标记的热点图中的每个点(i,j)到pl的距离,σ2是该高斯分布的标准差,用l1表示关节点对应的热点图的损失函数,用l2表示关节点连线对应的热点图的损失函数:

其中,X是输入图像,y是真实标记的关节点,p是相邻真实标记的关节点的连线,D是整个训练数据集,(i,j)表示热点图中的每个像素点的坐标,k和l分别表示y的序号和p的序号,Hi,j,k(X,w)表示神经网络预测输出的相应的第k关节点的热点图,Pi,j,l(X,w)表示神经网络预测输出的相应的第1关节点连线的热点图,w表示卷积神经网络的参数,结构指导卷积神经网络的总的损失函数loss1为:

loss1=l1+l2

3.根据权利要求1所述的一种基于结构指导深度学习的人体姿态估计方法,其特征在于,所述步骤3进一步具体为:将结构指导卷积神经网络的低层级特征层和高层次特征层进行融合,送入另外一支卷积神 经网络中进行学习和预测,低层次特征层包含更多的图像细节,高层次特征层则提供预测的初步结果;结构指导神经网络引入了人体结构约束信息,提供给融合卷积神经网络一个初步结果,并指导融合卷积网络的学习,融合卷积神经网络的目标函数为损失函数loss2为:

loss2=l1

整个神经网络总的损失函数losst为:

losst=loss1+λloss2

λ是平衡loss1和loss2的常数。

4.一种基于结构指导深度学习的人体姿态估计系统,其特征在于,包括如下模块:

数据采集模块,主要是通过相机得到包含有人体的图片;

计算机处理模块,包括图像预处理子模块,结构指导卷积神经网络子模块,融合卷积神经网络子模块,定位关节点子模块和可视化子模块。

当前第2页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1