来自 Facebook AI 和美国圣母大学的研究者提出了一种 6 自由度的实时 3D 人脸姿态估计技术,可以不依赖人脸检测和人脸关键点定位独立运行。
人脸检测是给照片中的每个人脸指定一个边界框,人脸关键点检测则需要定位特殊的人脸特征,如眼睛中心、鼻尖等。基于二者的两步走方法是很多人脸推理任务的关键所在,如 3D 重建。
这种方法的处理过程通常可以表述为:首先执行人脸检测,然后在每个检测到的人脸边界框中执行关键点检测。接下来,将检测到的关键点与参考 2D 图像或 3D 模型上对应的理想位置进行匹配,然后使用标准方法求解对齐变换。因此,「人脸对齐」和「关键点检测」这两个术语有时可以互换使用。
这种方法应用起来非常成功,但计算成本很高,尤其是那些 SOTA 模型。而且,关键点检测器通常针对由特定人脸检测器生成的边界框特性进行优化,因此一旦人脸检测器更新,关键点检测器就需要重新进行优化。最后,在下图 1 所示的密集人脸图像场景中,准确定位标准的 68 个人脸关键点会变得非常困难,进而加大了估计其姿态和人脸对齐的难度。
为了解决这些问题,来自 Facebook AI 和圣母大学的研究者提出了以下重要观察结果:
他们提出了一个易于训练的新型实时 6DoF 3D 人脸姿态估计解决方案,可以跳过人脸检测步骤,即使在非常拥挤的图片中也不例外(如图 1)。该方法在一个基于 Faster R-CNN 的框架中 regress 6DoF 姿态。
这项研究的创新之处在于,它真正摆脱了人脸对齐和关键点检测。「我们观察到,估计人脸的 6DoF 刚性变换比人脸目标点检测要简单。此外,6DoF 提供的信息要比人脸边界框标签丰富,」研究者解释道。
新方法的 pipeline 可以描述为:给定一张包含多张人脸的图像,首先估计每张人脸的 6DoF 姿态。由于 6DoF 人脸姿态可以转换为一个外在相机矩阵,进而将 3D 人脸映射到 2D 图像平面,因此预测得到的 3D 人脸姿态也可用于获取准确的 2D 人脸边界框。因此,人脸检测将成为这个过程的副产品,计算开销达到最小。
利用 6DoF 姿态估计替代人脸边界框检测之后,输入图像中所有的 3D 人脸形状都可以得到对齐。而且,由于该姿态将具有已知几何形状的 3D 形状与图像中的面部区域对齐,因此我们可以根据大小和形状调整生成的面部边界框,匹配特定的研究需求。
研究者使用小而快的 ResNet-18 骨干网络构建 img2pose 模型,并在 WIDER FACE 训练集上进行训练(该数据集包含弱监督标签和人工标注的真值姿态标签)。他们在两个领先的基准数据集( AFLW2000-3D 和 BIWI )上测试了 img2pose 的实时推断能力,发现该模型在实时运行时性能超越当前最优的人脸姿态估计器,还在关键点检测方面超越了具备类似复杂度的模型,尽管新模型并没有在边界框标签上进行优化。
以下是论文中的一些实验结果:
论文链接:https://arxiv.org/pdf/2012.07791.pdf
项目链接:https://github.com/vitoralbiero/img2pose
END