2D 图片变 3D:最新 3D GAN 可生成三维几何数据,模型速度提升 7 倍

2D 图片变 3D,还能给出 3D 几何数据?

英伟达和斯坦福大学联合推出的这个 GAN,真是刷新了 3D GAN 的新高度。

而且生成画质也更高,视角随便摇,面部都没有变形。

与过去传统的方法相比,它在速度上能快出 7 倍,而占用的内存却不到其十六分之一

最厉害的莫过于还可给出 3D 几何数据,像这些石像效果,就是根据提取的位置信息再渲染而得到的。

甚至还能实时交互编辑。

该框架一经发布,就在推特上吸引了大量网友围观,点赞量高达 600+。

怎么样?是不是再次刷新你对 2D 升 3D 的想象了?

显隐混合 + 双重鉴别

事实上,只用一张单视角 2D 照片生成 3D 效果,此前已经有许多模型框架可以实现。

但是它们要么需要计算量非常大,要么给出的近似值与真正的 3D 效果不一致。

这就导致生成的效果会出现画质低、变形等问题。

为了解决以上的问题,研究人员提出了一种显隐混合神经网络架构 (hybrid explicit-implicit network architecture)。

这种方法可以绕过计算上的限制,还能不过分依赖对图像的上采样。

从对比中可以看出,纯隐式神经网络(如 NeRF)使用带有位置编码(PE)的完全连接层(FC)来表示场景,会导致确定位置的速度很慢。

纯显式神经网络或混合了小型隐式解码器的框架,虽然速度更快,但是却不能保证高分辨率的输出效果。

而英伟达和斯坦福大学提出的这个新方法 EG3D,将显式和隐式的表示优点结合在了一起

它主要包括一个以 StyleGAN2 为基础的特征生成器和映射网络,一个轻量级的特征解码器,一个神经渲染模块、一个超分辨率模块和一个可以双重识别位置的 StyleGAN2 鉴别器。

其中,神经网络的主干为显式表示,它能够输出 3D 坐标;解码器部分则为隐式表示。

与典型的多层感知机制相比,该方法在速度上可快出 7 倍,而占用的内存却不到其十六分之一。

与此同时,该方法还继承了 StyleGAN2 的特性,比如效果良好的隐空间(latent space)。

比如,在数据集 FFHQ 中插值后,EG3D 的表现非常 nice:

该方法使用中等分辨率(128 x 128)进行渲染,再用 2D 图像空间卷积来提高最终输出的分辨率和图像质量。

这种双重鉴别,可以确保最终输出图像和渲染输出的一致性,从而避免在不同视图下由于卷积层不一致而产生的问题。

▲ 两图中左半边为最终输出效果,右半边为渲染输出

而没有使用双重鉴别的方法,在嘴角这种细节上就会出现一些扭曲。

▲ 左图未使用双重鉴别;右图为 EG3D 方法效果

数据上,与此前方法对比,EG3D 方法在 256 分辨率、512 分辨率下的距离得分(FID)、识别一致性(ID)、深度准确性和姿态准确性上,表现都更好。

团队介绍

此项研究由英伟达和斯坦福大学共同完成。

共同一作共有 4 位,分别是:Eric R. Chan、Connor Z. Lin、Matthew A. Chan、Koki Nagano。

其中,Eric R. Chan 是斯坦福大学的一位博士研究生,此前曾参与过一些 2D 图像变 3D 的方法,比如 pi-GAN。

Connor Z. Lin 是斯坦福大学的一位正在读博二的研究生,本科和硕士均就读于卡内基梅隆大学,研究方向为计算机图形学、深度学习等。

Matthew A. Chan 则是一位研究助理,以上三人均来自斯坦福大学计算机成像实验室(Computational Imaging Lab)。

Koki Nagano 目前就职于英伟达,担任高级研究员,研究方向为计算机图形学,本科毕业于东京大学。

论文地址:

https://arxiv.org/abs/2112.07945

参考链接:

https://matthew-a-chan.github.io/EG3D/

Published by

风君子

独自遨游何稽首 揭天掀地慰生平

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注