Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields
1、主要成果
提出将人体部位与个体联系起来的非参数化方法—PAFs,实现自底向上的实时多人姿态估计,在性能和效率上均实现当时最优的结果。
2、算法细节
2.1、算法原理
首先预测一组图像中人体部位位置的 2D 置信图 S,以及一组表示部位相似性的 2D 矢量场 L,它们表示部位之间的关联程度。集合 S=(S1,S2,…,SJ) ,Sj∈Rw×h,j∈{1…J}。集合 L=(L1,L2,…,LC) 有 C 个矢量场,每个肢体一个。其中,Lc∈Rw×h×2,c∈{1…C},Lc 中的每个图像位置编码一个 2D 向量。最后,通过贪婪推理解析置信度图和亲和场,以输出图像中所有人的 2D 关键点。
2.2、实现细节
2.2.1、网络结构
- 使用 VGG−19 前 10 层提取图像特征 F
- 将 F 输入到 stage−1 置信图分支网络和亲和场分支网络,分别输出 S1=ρ1(F) 和 L1=ϕ1(F)
- 重复 stage−1,输入为前一阶段的 St 和 Lt 以及图像特征 F 的 Concatenate。即,StLt=ρt(F,St−1,Lt−1),∀t≥2=ϕt(F,St−1,Lt−1),∀t≥2
2.2.2、损失函数
为每一阶段的两个分支网络分别设计损失函数,对损失进行空间加权,来解决一些数据集不能完全标记所有人的问题。
fStfLt=j=1∑Jp∑W(p)⋅∥∥Sjt(p)−Sj∗(p)∥∥22=c=1∑Cp∑W(p)⋅∥∥Lct(p)−Lc∗(p)∥∥22
其中,W 为二进制 mask,当图像位置 p 没有注释时,W(p)=0,避免训练时惩罚 TP 预测。每个阶段的中间监督通过定期补充梯度来解决梯度消失问题。总损失为:
f=t=1∑T(fSt+fLt)
2.2.3、GT 置信图生成
GT 置信图根据标注的人体关键点位置生成,以多人为例,第 k 个人的第 j 个可见部位的置信图表示为 Sj,k∗:
Sj,k∗(p)=exp(−σ2∥p−xj,k∥22)
其中,p∈R2,xj,k 为关键点标注位置。则,
Sj∗(p)=kmaxSj,k∗(p)
2.2.4、PAFs 生成
Lc,k∗(p)={v0 if p on limb c,k otherwise
当点 p 在 k 的肢体 c 上时,Lc,k∗(p) 等于从关键点 j1 指向关键点 j2 的单位向量。否则为零向量。肢体上的点必须满足以下条件:
0≤v⋅(p−xj1,k)≤lc,k and ∣v⊥⋅(p−xj1,k)∣≤σl
其中肢体长度:lc,k=∥xj2,k−xj1,k∥2
肢体宽度:σl 为像素距离
最终 PAFs 为图像中所有人的平均亲和场:
Lc∗(p)=nc(p)1k∑Lc,k∗(p)
其中,nc(p) 为点 p 处所有人产生的非零向量个数。
测试阶段,通过沿着连接候选部位位置的线段,计算相应 PAF 上的线积分来测量候选部位检测之间的关联。
E=∫u=0u=1Lc(p(u))⋅∥dj2−dj1∥2dj2−dj1du
p(u) 为两个人体部位 dj1 和 dj2 的位置插值。
p(u)=(1−u)dj1+udj2
实际中通过对 u 均匀采样,然后求和来近似积分。
2.2.5、PAFs 多人分析
通过对检测置信度图执行非最大值抑制,以获得一组离散的部位候选位置。每个部位可能生成多个候选,经过组合生成大量的候选肢体。通过在 PAF 上计算线积分来对每个候选肢体打分。找到最优解的问题对应于一个已知为 NP 难的 K 维匹配问题。
在这篇文章中,作者提出了一种贪婪的简化方法,能够始终产生高质量的匹配。推测原因是成对关联分数隐含地编码了全局上下文,这是由于 PAF 网络的感受野很大。
DJ={djm:j∈{1…J},m∈{1…Nj}}
djm 表示部位 j 的第 m 个候选。接下来需要确定部位对构成的肢体。zj1j2mn∈{0,1}表示 dj1m 和 dj2n 是否相连。目标是找到最优分配方案:
Z={zj1j2mn:j1,j2∈{1…J},m∈{1…Nj1},n∈{1…Nj2}}
寻找最佳关联简化为最大权重二部图匹配问题。该问题中,结点为 Dj1 和 Dj2,边是两组结点间所有可能的连接。每条边的权重为之前计算的线积分 E。二部图中的匹配是以没有两条边共享一个结点的方式选择的边的子集。优化的目标是为选定的边找到最大权重的匹配:
ZcmaxEc=Zcmaxm∈Dj1∑n∈Dj2∑Emn⋅zj1j2mn
s.t. ∀m∈Dj1,n∈Dj2∑zj1j2mn≤1∀n∈Dj2,m∈Dj1∑zj1j2mn≤1
通过 Hungarian 算法获取最优匹配。
当涉及到寻找多人的全身姿态时,确定 Z 是一个 K 维匹配问题。这个问题是 NP 难并且有许多松弛存在。作者为优化添加两个松弛。
- 选择最小数量的边来获得人体姿态的生成树骨架,而不是使用完整的图,如图 6c 所示。
- 进一步将匹配问题分解成一组二部匹配子问题,并独立地确定相邻树节点中的匹配,如图 6d 所示。
结果表明最小贪婪推理很好地近似了全局解,只需要计算成本的一小部分。原因是相邻树节点之间的关系是由 PAFs 显式建模的,但在内部,不相邻树节点之间的关系是由 CNN 隐式建模的。这一特性的出现是因为 CNN 有一个大的感受野,并且来自非相邻树节点的 PAFs 也影响预测的 PAF 。因此,优化可以被简化为:
ZmaxE=c=1∑CZcmaxEc
因此,可以独立地获得每种肢体类型的肢体连接候选。对于所有肢体连接候选,可以将共享相同部位检测候选的连接,组合成多人的全身姿势。在树结构上的优化方案比在全连接图上的优化快几个数量级。
2.2.6、结果
3、总结
主要贡献——PAFs:
-
部位置信图生成:对图像中所有人的同一类部位的高斯分布置信图取上限得到
-
亲和场生成:对图像中所有人的同一类肢体的亲和场求平均得到
-
预测结果生成:
- 将所有部位全连接图简化为树状图(类似传统方法的姿态树)
- 树形图也嫌麻烦,再拆分为二部图优化,每次解决一类肢体的生成问题
- 通过匈牙利算法求解,获得最大权重匹配的节点分配策略
-
边权重的确定:沿着部位位置连接成的线段,计算 PAF 上的线积分即为权重
4、感想
主要是贡献—PAFs,论文中涉及的具体的深度学习的实现细节不多,感觉这一块有一些提升的空间,可以尝试结合深度学习网络设计、损失函数设计方面的一些新的方法。