近日,英伟达发布了最新的GET3D模型,能快速生成虚拟世界的对象,而且只需要一块GPU每秒就能产出大约20个模型。

Abracadabra!

在2D生成3D模型上,英伟达亮出了自称「世界级」研究:GET3D。

通过2D图像训练后,该模型可生成具有高保真纹理和复杂几何细节的3D形状。

究竟有多厉害?

形状、纹理、材质自定义

GET3D之所以得名,是因为它能够生成显式纹理3D网格(Generate Explicit Textured 3D meshes )。

也就是说,它创建的形状是三角形网格的形式,就像纸模型一样,上面覆盖着纹理材质。

关键是,这个模型可以生成多种多样,且高质量的模型。

比如,椅子腿上的各种轮子;汽车的车轮,灯和车窗;动物的耳朵、角;摩托车的后视镜,车轮胎上的纹理;高跟鞋,人类衣服…

街道两旁的独特建筑,不同的车辆呼啸而过,还有不同的人群穿梭而过…

若想通过手动建模做出同样的3D虚拟世界非常耗时。

尽管先前的3D生成AI模型虽然比手动建模更快,但它们在生成更多丰富细节模型的能力还是欠缺。

即便是,最新的逆向渲染方法也只能基于从各个角度拍摄的2D图像生成3D对象,开发人员一次只能构建一个3D物体。

GET3D可就不一样了。

开发者可以轻松地将生成模型导入到游戏引擎、3D建模器和电影渲染器中,对它们进行编辑。

当创建者将GET3D生成的模型导出到图形应用程序,他们就可以在模型所在的场景中移动或旋转时应用逼真的照明效果。

如图所示:

另外,GET3D还可以做到文本引导形状生成。

通过使用英伟达的另一个AI工具StyleGAN-NADA,开发人员可以使用文本提示为图像添加特定的风格。

比如,可以将渲染后的汽车变成一辆烧毁的汽车或出租车

将一个普通的房子改造成砖房、着火的房子,甚至是鬼屋。

或者将老虎纹、熊猫纹、还有《辛普森一家》的特色应用在任何动物身上…

简直就是辛普森一家的「动物森友会」…

英伟达介绍,在单个英伟达GPU上训练,GET3D每秒可以生成大约20个对象。

在此,它所学习的训练数据集越大、越多样化,输出的多样性和详细程度就越高。

英伟达称,研究团队仅用2天时间,就使用A100 GPU在大约100万张图像上训练了模型。

研究方法与过程

GET3D框架,主要作用是合成有纹理的三维形状。

生成过程分为两个部分:第一部分是几何分支,可以输出任意拓扑结构的表面网格。另一部分则是纹理分支,它会产生一个纹理场,由此可以在表面点上进行查询。

在训练过程中,一个可微分光栅化器被用来高效渲染所产生的纹理网格,并渲染成二维的高分辨率图像。整个过程是可分的,允许通过传播二维判别器的梯度,从图像中进行对抗性训练。

之后,梯度从二维判别器传播到两个发生器分支。

研究人员又进行了广泛的实验来评估该模型。他们首先将GET3D生成的三维纹理网格的质量与现有的使用ShapeNet和Turbosquid数据集生成的做比较。

接下来,研究人员根据对比结果在之后的研究中对模型进行了优化,并做了更多实验。

GET3D模型在几何形状和纹理能够实现相分离。

如图,在每一行中展示了由相同的几何隐藏代码生成的形状,同时更改了纹理代码。

在每一列中展示了由相同的纹理隐藏代码生成的形状,同时更改了几何代码。

另外,研究人员在每一行中相同的纹理隐藏代码生成的形状,从左到右插入几何隐藏代码。

并由相同的几何隐藏代码生成的形状,同时从上到下插入纹理代码。结果显示,每个插值对生成模型都是有意义的。

在每个模型的子图中,GET3D能够在所有类别的不同形状之间的生成实现平滑过渡。

在每一行中,通过添加一个小噪声来局部扰乱隐藏代码。通过这种方式,GET3D能够在局部生成外观相似但略有差异的形状。

研究人员指出,未来版本的GET3D可以使用摄像机姿态估计技术,让开发人员训练模型的现实世界的数据,而不是合成数据集。

未来,通过改进,开发人员可以在各种3D形状上一次性训练GET3D,而不需要一次在一个对象类别上训练它。

英伟达人工智能研究副总裁Sanja Fidler表示,

GET3D让我们离人工智能驱动的3D内容创作大众化又近了一步。它即时生成带纹理3D形状的能力可能会改变开发人员的游戏规则,帮助他们用各种有趣的对象快速填充虚拟世界。
作者介绍

论文一作Jun Gao是多伦多大学机器学习小组的博士生,导师是Sanja Fidler。

除了学历优异以外,他还是英伟达多伦多人工智能实验室的研究科学家。

他的研究主要集中在深度学习(DL),目标直指结构化几何表示学习。同时,他的研究还从人类对2D和3D图像、视频的感知中获得见解。

这么一位优秀的高材生,来自北京大学。他于2018年本科毕业,获得学士学位。在北大期间,他和王立威教授一同工作。

毕业后他还在斯坦福大学、MSRA和英伟达进行过实习。

Jun Gao的导师同样是业内翘楚。

Fidler是多伦多大学的副教授,Vector研究所的一名教师,同时,她还是该研究所的联合创始成员之一。

教学之外,她还是英伟达公司的人工智能研究副总裁,领导着多伦多的一个研究实验室。

在来到多伦多之前,她是芝加哥丰田技术研究所的研究助理教授。该研究所位于芝加哥大学校园内,算是个学术机构。

Fidler的研究领域集中在计算机视觉(CV)和机器学习(ML),聚焦于CV和图形学的交叉领域、三维视觉,以及三维重建与合成,还有图像注释的互动方法等等。