选自github
作者：Pramook Khungurn
机器之心编译
参与：Panda、张倩
有多少人想要当个主播却因为害羞或对自己颜值不自信而放弃？也许虚拟主播和变声器能帮你实现梦想，但实际做起来却要困难得多。现在，Google Japan 的软件工程师 Pramook Khungurn 开发了一套基于深度学习的系统，只需一张正脸的动漫人物图，就能轻松合成各种面部和头部动作的动画，搭配人脸跟踪器等应用，能让人以很低的成本轻松制作动画主播节目。
项目作者最喜欢的虚拟主播之一——Shirakami Fubuki。
我训练了一个可将动漫人物的脸做成动画的网络。
这种新网络能完成这样的工作：
我将该系统连接到了一个人脸跟踪器上。从而让人物可以模仿我的头部动作：
我也可以迁移已有视频中的人脸运动：
这个网络的输入是人物的正面图像，人物的姿势由 6 个数值指定。
我通过渲染 8000 个可下载的动漫人物的 3D 模型创造了一个新的数据集。
我使用的方法结合了之前两项研究。一是 Pumarola et al. 2018 年的 GANimation 论文《GANimation: Anatomically-aware Facial Animation from a Single Image》，我将其用于修改面部的特征（具体来说是闭上眼睛和嘴）。二是 Zhou et al. 2016 年根据外观流实现目标旋转的论文《View Synthesis by Appearance Flow》，我将其用于实现人脸的旋转。
概览
我想要解决的问题是这样的：给定某个动漫人物的一张人脸图像和一个「姿势（pose）」，生成同一人物的另一张图像，并且其人脸会根据姿势而变化。在这里，姿势是一个指定了该人物面部表情和头部旋转情况的数值的集合。具体来说，我的姿势有 6 个数值，分别对应前面动图中的不同滑块。我将在「问题设定」一节讨论输入和输出的细节。
该系统的输入和输出
我使用了深度学习来解决这样的一个问题。这需要我解答下面这两个问题：
我要使用什么数据来训练这个网络？
我要使用什么网络架构以及具体采用怎样的训练方法？
事实证明，第一个问题是最主要的挑战。我需要一个包含姿势标签的人脸图像数据集。Emotionet 是一个包含所需类型的标签的大型人脸数据集。
但是，就我所知，还没有类似的动漫人物数据集。因此，我专门为本项目生成了一个新的数据集。
为此，我利用了这一事实：网上就有数以万计可下载的动漫人物 3D 模型，这些模型是用一款名为 MikuMikuDance 的 3D 动画软件创造的。我下载了大约 8000 个模型，然后使用了它们来渲染随机姿势的动漫人脸。详细的数据准备步骤将在「数据集」一节介绍。
我根据实现 3D 人物模型动画化的方法设计了该网络。我将这样的一个过程分成了两步。第一步是改变面部表情；即控制眼睛和嘴的开闭程度。第二步是人脸旋转。
我为每一步都使用了一个单独的网络，并将第一个网络的输出用作第二个网络的输入。我将第一个网络称为人脸变形器（face morpher），第二个网络称为人脸旋转器（face rotator）。
该系统的执行过程分为两个步骤
人脸变形器使用了 Pumarola et al. 的 ECCV 2018 论文中所用的生成器架构。
该网络可通过生成另一张包含对原始图像的修改的新图像来改变人脸表情。这些改变会通过一个 α 掩码与原图像组合，而这个 α 掩码也是由该网络本身生成的。我发现，这种架构在修改图像中的小部件方面表现非常出色：这里是闭上眼睛和嘴。
人脸旋转器则要复杂得多。我在单个网络中实现了两个算法来旋转人脸，因此该网络有两个输出。这两个算法是：
Pumarola et al. 的算法。这就是刚刚用来修改面部表情的算法，但现在该网络的任务是旋转人脸。
Zhou et al. 的视图合成算法。他们的目标是旋转图像中的 3D 目标。他们的实现方法是让神经网络计算一个外观流（appearance flow）：这是一个映射图，能指示输出图像中的每个像素应该复制输入中哪个像素的颜色。
外观流能得到保留了原有纹理的清晰结果，但这种方法不善于描绘在旋转后变得可见的部分。另一方面，Pumarola et al 的架构得到的结果较模糊，但能描绘被遮挡的部分，因为其训练目标就是改变原始图像的像素，而不是从已有图像复制像素。为了将这两种方法的长项结合到一起，我训练了另一个网络通过一个 α 掩码将这两个输出融合到一起。这个网络还能输出「修整（retouch）」图像，即可将组合后的图像与另一个 α 掩码再组合起来。
网络系统，这里更详细地展示了人脸旋转器
我将在「网络」一节详细讨论所有网络的架构和训练方式。
问题设定
如前所言，人物的人脸配置是通过姿势控制的。这里的姿势是一个 6 维向量，其中三个分量控制人脸特征，其取值范围为闭区间 [0,1]。
在这三个分量中，两个分量控制眼睛的睁闭：一个控制左眼，一个控制右眼。值为 0 时表示眼睛完全睁开，为 1 时表示完全闭眼。
另一个分量控制嘴的开合。但是，这个量为 0 时表示嘴完全闭上，为 1 时表示嘴完全张开。眼睛和嘴参数的语义互相抵触是因为这些 3D 模型的变形（morph）权重的语义就是如此：https://en.wikipedia.org/wiki/Morph_target_animation
6 维姿势向量中另外三个分量控制的是头部的旋转方式。用 3D 动画术语来说，头部是由一个「骨骼（bone）」连接的两个「关节（joint）」控制的。颈根（neck root）关节是脖子与身体相连的位置，颈尖（neck tip）关节则是脖子与头相连的位置。在人物骨架中，颈尖是颈根的延展。因此，应用于颈根的 3D 变换也会对颈尖产生影响，但反过来却不会。
控制角色头部的两个关节
概括起来，输入是人物人脸的图像与一个 6 维姿势向量。输出是另一张姿势根据情况有所调整的人脸图像。
网络
如前所述，我的神经网络包含多个子网络。下面详细的介绍它们。
人脸变形器
确定人物人脸的姿势的第一步是修改其人脸特征。更具体来说，我们应该人物能闭上眼睛和嘴。
Pumarola et al. 在他们的论文中描述了一种能够准确的通过给定的动作单位（AU：Action Units，表示面部肌肉的运动）修改人脸特征的网络。
因为 AU 是非常通用的编码系统，所以他们的网络可不止能闭眼和闭嘴。因此，我觉得这能有效地处理我们手头上的任务。它没有让我失望。
但是，我没有使用该论文的所有方法，因为我的问题比他们的简单很多。尤其是他们的训练数据没有来自同一个人的表情不同的人脸对。因此，他们需要用带有循环一致性损失的 GAN 来执行无监督学习。但是，我的数据是配对好的，所以我可以执行简单的监督学习。所以，我只需要他们的生成器网络。
下面我详细说说 Pumarola et al. 的生成器。下图是其架构示意图。
人脸变形器的架构。这是针对本案例对 Pumarola et al. 的论文中图 3 的复现
该网络会通过生成改变图像来实现对面部表情的修改，这个改变图像是原图像与一个 α 掩码（Pumarola et al. 称之为注意力掩码，但我这里使用了一个更常用的术语）的组合。要做到这一点，输入图像和姿势会被送入一个编码器-解码器网络，这会为输入图像的每个像素得到一个 64 维的特征向量。然后，这些特征向量构成的图像会经过两次不同的 2D 卷积单元和适当非线性的训练处理，得到 α 掩码和改变图像。附录 A.1. 给出了该网络的详细指标。
Pumarola et al. 使用了一个相对复杂的损失函数来训练他们的网络。让我惊讶的是，对我的问题来说，简单的 L1 像素差异损失就足够了。这个损失函数的数学描述如下：
我使用了与 Pumarola et al. 一样设置的 Adam 算法来优化网络：学习率 1^ 4、β_1=0.5、β_2=0.999，批大小为 25。网络训练了 6 epoch（3 000 000 个样本），在我的 GeForce GTX 1080 Ti GPU 上耗时两天。
人脸旋转器
人脸旋转器由两个子网络构成。双算法旋转器使用两个不同的算法旋转人物的面部，其中每个算法都有各自的优势和短板。为了将两者的优势结合到一起，还有一个组合器（combiner）能使用一个 α 掩码将两个算法输出的图像融合到一起，并且还能对图像进行修整，实现质量提升。
双算法旋转器
下图描述了其架构，附录 A.2. 给出了详细说明。
可以将该网络看作是对 Pumarola et al. 的生成器的扩展：其包含该生成器的所有单元，还多了一个新的输出路径。原有的路径的任务只是旋转人脸，而不是闭上眼睛和嘴。新的路径则是使用了 Zhou et al. 论文中描述的方法来旋转目标。
其中的思想是：物体旋转（尤其是角度较小时）很大程度上涉及到将输入图像内的像素移动到不同位置。因此 Zhou et al. 提出计算外观流：这是一种映射图，能指示输出图像中的每个像素应该复制输入中的哪个像素。然后，这个映射图和原始图像会被传递给一个像素采样单元以生成输出图像。在我的架构中，外观流就是简单地通过将编码器-解码器网络的输出传递给一个新的卷积单元而计算的。
我在训练网络时使用了两个不同的损失。第一个就是简单的 L1 像素差异损失：
第二个损失是 L1 像素差异损失和 Johnson et al. 的感知特征重建损失的和：
同样，优化用的是 Adam，使用了与人脸变形器一样的超参数。一共训练了 6 epoch（3 000 000 个样本）。当使用 L1 损失时，我将批大小设为 25，训练同样用去了大约两天时间。但是，由于评估特征重建损失需要远远更大的内存，所以当使用这种感知损失进行训练时，我必须将批大小降至 8。在这种情况下训练也用去了 6 天。
组合器
看看下面双算法旋转器的输出，能够正常的看到单独任何一个算法的结果都不够好。
使用不同算法将人物的脖子向右旋转 15°
上图展示了对人物脖子进行旋转操作的情况，旋转后她一部分原本被遮挡的长发会变得可见。能够正常的看到，Pumarola et al. 的算法会得到模糊的人脸。我推测原因是该网络需要根据压缩的特征编码生成所有新像素，这可能会丢失原图像中的高频细节。之前其它一些编码器-解码器架构的研究也观察到了类似的行为，比如 Tatarchenko et al. 和 Park et al.。
Zhou et al. 则是复用输入图像的像素，因此能够产生清晰的结果。然而，通过复制已有像素难以重建之前被遮挡的部分，尤其是所要复制的位置距离很远时。在上图 b 中能够正常的看到，Zhou et al. 的算法使用了手臂的像素来重建去除遮挡后的头发。另一方面，Pumarola et al. 方法得到的头发颜色更自然。
通过组合这两个算法的输出，我们大家可以得到好得多的结果：可见像素调整位置后会保持原有的清晰度，去除遮挡部分的像素也能获得自然的颜色。下图展示了组合器网络的架构，详情参阅附录 A.3.
组合器的架构
这个组合器网络的主体是 U-Net，这有助于对每个像素做相关操作。然后它的输出会被变换成两个 α 掩码和一张变化图像。其中第一个 α 掩码会被用于组合两张输入图像。然后，第二个 α 掩码和变化图像则会与前一步的输出组合，得到最终结果。最后一步是对组合得到的图像进行修正，以提升其质量。
为了减少内存用量，这个组合器是与双算法旋转器分开训练的。我在所有训练样本上运行了后者，从而生成前者的输入。同样，我实验了两个损失函数。第一个是 L1 损失。
第二个是感知损失：
这个训练流程类似于人脸变形器的训练过程。但是，为了方便，这个训练仅进行了 3 epoch，而非 6 epoch。使用 L1 损失时批大小为 20，训练消耗了一天时间。使用感知损失时，批大小为 12，训练持续了两天时间。
评估
定量评估
表现评估使用了两个指标。第一个是网络的输出与基本真值图像之间的平均每像素均方根误差（RMSE）。第二个是平均结构相似度指数（SSIM）。分数使用测试数据集中的 10 000 个样本计算得到。
能够正常的看到，仅使用一个人脸旋转路径的表现通常比组合使用两个路径的表现差。一个值得关注的例外是 PU-P，其在 RMSE 上的表现优于除 FU-P-P 之外的所有网络。但是在 SSIM 指标上，组合的效果总是更好。
还可以观察到另一个趋势：使用感知损失时通常在两个指标上都表现更好。但是，在 SSIM 指标上表现最佳的配置是 FU-P-L1 而非 FU-P-P。
看起来最佳的配置是 FU-P-L1 和 FU-P-P，这两者在这两个指标上都取得了第一和第二名的成绩。因此，我们应该进一步检查生成的图像才能确定哪个更好。
定性评估
下面展示由一种配置生成的结果。
由 FU-P-P 网络配置渲染的人物动画
我们从视觉质量方面比较一下各种网络配置。PU-L1 和 PU-P 得到的结果过于模糊，质量差。这表明虽然 Pumarola et al. 的架构能有效地修改面部上的小组件，但当需要修改图像中较大部分时，其效果并不好。还可以观察到，由于使用了感知损失，所以 PU-P 得到的结果更清晰。但是，这种损失的副作用是会产生棋盘状伪影。
PU-L1 和 PU-P 生成的图像
ZH-L1 和 ZH-P 则能得到非常清晰的结果，因为它们是直接复制输入图像的像素。但是，它们可能生成会让人物变样的不规则伪影。
ZH-L1 和 ZH-P 生成的图像
对于使用了所有子网络的配置，面部和身体的大部分区域都很清晰，因为这些像素是由组合器从 Zhou et al. 的路径选取的。因为 Pumarola et al. 的路径比 Zhou et al. 的更不容易复制近邻像素，所以组合器可以从前者的像素中做出合理的选择，很大程度上（但并非完全）消除后者产生的扰动伪影。因此，完全配置的网络比仅使用一条路径的网络能得到质量更好的图像。但是，这些输出中消除遮挡的部分依然很模糊。下图表明，在 4 种完全的配置中，FU-P-P 得到的结果最清晰。但是，某些人（包括我）可能并不喜欢棋盘状伪影，而更偏爱 FU-P-L1 的更平滑的输出。
FU-L1-L1、FU-L1-P、FU-P-L1、FU-P-P 生成的图像
我注意到所有网络似乎对人物的身体结构都有某些特定的程度的高层面理解。举个例子，大和伊织（Yamato Iori）的右眼被头发挡住了，当她闭眼时，任何网络都不会去移动她的头发。但是，我们也观察到一些因为图像解析不正确而导致的失败案例。比如夜樱多摩（Yozakura Tama）有一条挂在身前的长发辫。所有网络配置都将其看作是两部分，在移动头时仅移动了其中的上部分。衣服和配饰也可能被误认为是头部的一部分，见下图。
由于图像解析不正确而导致的失败案例
总的来说，FU-P-L1 和 FU-P-P 应该是最佳的网络配置，因为它们能在不产生大规模伪影的情况下生成基本上清晰的输出。FU-P-P 生成的图像更清晰，但却有棋盘状伪影，FU-P-L1 得到的结果更模糊但更平滑。但这两个网络在解析输入图像时都可能出错，从而得到身体结构/物理结构上不可信的动画。
绘制动画
本项目的最终目标是将非 3D 渲染的图画变成动画。我使用最佳的网络 FU-P-P 评估了我的方法的表现，评估使用了 Nijisanji 旗下虚拟主播的动漫图像和 Waifu Labs 生成的人物。下面展示了 16 个人物的动画：
因为网络是在渲染过的动画人物上训练的，所以他们能很好地应用于同一艺术风格的绘画。需要指出，网络在处理眼睛方面的表现尤其好。对于大多数人物，他们都能正确地闭上眼睛，即使眼的一部分被头发遮挡时也是如此。
GANimation 论文：https:///research/GANimation/index.html
根据外观流实现目标旋转的论文：https://arxiv.org/abs/1605.03557
完整版博客介绍：https://pkhungurn.github.io/talking-head-anime/
机器之心「SOTA模型」：22大领域、127个任务，机器学习 SOTA 研究一网打尽。
