Unsupervised generation of high-quality multi-view-consistent images and 3D shapes using only collections of single-view 2D photographs has been a long-standing challenge. Existing 3D GANs are either compute-intensive or make approximations that are not 3D-consistent; the former limits quality and resolution of the generated images and the latter adversely affects multi-view consistency and shape quality. In this work, we improve the computational efficiency and image quality of 3D GANs without overly relying on these approximations. For this purpose, we introduce an expressive hybrid explicit-implicit network architecture that, together with other design choices, synthesizes not only high-resolution multi-view-consistent images in real time but also produces high-quality 3D geometry. By decoupling feature generation and neural rendering, our framework is able to leverage state-of-the-art 2D CNN generators, such as StyleGAN2, and inherit their efficiency and expressiveness. We demonstrate state-of-the-art 3D-aware synthesis with FFHQ and AFHQ Cats, among other experiments.
The full architecture and related paper can be found at https://nvlabs.github.io/eg3d/.
Full instructions on training the models can be found at https://github.com/NVlabs/eg3d.
FFHQ (512x512): We realigned and recropped the original FFHQ dataset, with light augmentations.
FFHQ-Rebalanced (512x512): On top of our previous cropping and augmentations for FFHQ, we also include several duplications of images for less-frequently seen poses.
AFHQv2 (512x512): We use the cats-train subset of the AFHQv2 dataset released by stylegan3. We also apply further augmentation in training.
Shapenet Cars (128x128): We render randomly sampled views of many synthetic cars in the Shapenet dataset.
The datasets as well as more information can be found at https://github.com/NVlabs/eg3d.
Results, as well as estimated training duration vary greatly depending on available resources and training options. See the Readme for several examples and base recommendations for training.
Requirements:
Datasets are stored as uncompressed ZIP archives containing uncompressed PNG files and a metadata file dataset.json for labels. See dataset_tool.py for specific options and instructions for dataset creation.
See information and examples of the output at our project page here: https://nvlabs.github.io/eg3d/.
https://github.com/NVlabs/eg3d
Copyright (C) 2021-22, NVIDIA Corporation & affiliates. All rights reserved.
This work is made available under the Nvidia Source Code License.