Skip to content

jimmyYliu/Animatable-3D-Gaussian

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Animatable 3D Gaussian

Yang Liu*, Xiang Huang*, Minghan Qin, Qinwei Lin, Haoqian Wang (* indicates equal contribution)
| Webpage | Full Paper

Image text Abstract: Neural radiance fields are capable of reconstructing high-quality drivable human avatars but are expensive to train and render and not suitable for multi-human scenes with complex shadows. To reduce consumption, we propose Animatable 3D Gaussian, which learns human avatars from input images and poses. We extend 3D Gaussians to dynamic human scenes by modeling a set of skinned 3D Gaussians and a corresponding skeleton in canonical space and deforming 3D Gaussians to posed space according to the input poses. We introduce a multi-head hash encoder for pose-dependent shape and appearance and a time-dependent ambient occlusion module to achieve high-quality reconstructions in scenes containing complex motions and dynamic shadows. On both novel view synthesis and novel pose synthesis tasks, our method achieves higher reconstruction quality than InstantAvatar with less training time (1/60), less GPU memory (1/4), and faster rendering speed (7x). Our method can be easily extended to multi-human scenes and achieve comparable novel view synthesis results on a scene with ten people in only 25 seconds of training.

Prerequisites

  • Cuda 11.7
  • Conda
  • A C++14 capable compiler
    • Windows: Visual Studio 2019 or 2022
    • Linux: GCC/G++ 8 or higher

Setup

First make sure all the Prerequisites are installed in your operating system. Then, invoke

conda create --name anim-gaussian python=3.8
conda activate anim-gaussian
bash ./install.sh

Prepare Data

We use PeopleSnapshot and GalaBasketball datasets and correspoding template body model. Please download and organize as follows

|---data
|   |---Gala
|   |---PeopleSnapshot
|   |---smpl

Train

To train a scene, run

python train.py --config-name <main config file name> dataset=<path to dataset config file>

For the PeopleSnapshot dataset, run

python train.py --config-name peoplesnapshot.yaml dataset=peoplesnapshot/male-3-casual

For the GalaBasketball dataset, run

python train.py --config-name gala.yaml dataset=gala/idle

Test

To test the trained model performance on the test set, run

python test.py --config-name <main config file name> dataset=<path to dataset config file>

To animate the trained model at the novel poses, run

python test.py --config-name animate.yaml output_path=<path to train output> dataset=<path to animation dataset config file>

For example, to animate the model trained on the idle dataset using poses on the dribble dataset, run

python test.py --config-name animate.yaml output_path=idle dataset=gala/dribble

BibTeX

@article{liu2023animatable,
  title={Animatable 3D Gaussian: Fast and High-Quality Reconstruction of Multiple Human Avatars},
  author={Liu, Yang and Huang, Xiang and Qin, Minghan and Lin, Qinwei and Wang, Haoqian},
  journal={arXiv preprint arXiv:2311.16482},
  year={2023}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published