skhu101 / GauHuman

Code for our CVPR'2024 paper "GauHuman: Articulated Gaussian Splatting from Monocular Human Videos"
Other
285 stars 23 forks source link

GauHuman: Articulated Gaussian Splatting from Monocular Human Videos

Shoukang Hu; Ziwei Liu
S-Lab, Nanyang Technological University
CVPR 2024

GauHuman learns articulated Gaussian Splatting from monocular videos with both fast training (1~2 minutes) and real-time rendering (up to 189 FPS).
:open_book: For more visual results, go checkout our project page This repository will contain the official implementation of _GauHuman: Articulated Gaussian Splatting from Monocular Human Videos_.
## :mega: Updates [12/2023] Training and inference codes for ZJU-Mocap_refine and MonoCap are released. ## :desktop_computer: Requirements NVIDIA GPUs are required for this project. We recommend using anaconda to manage the python environments. ```bash conda create --name gauhuman python=3.8 conda activate gauhuman conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia pip install submodules/diff-gaussian-rasterization pip install submodules/simple-knn pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl pip install -r requirement.txt ``` Tips: We implement the [alpha mask loss version](https://github.com/ashawkey/diff-gaussian-rasterization) based on the official [diff-gaussian-rasterization](https://github.com/graphdeco-inria/diff-gaussian-rasterization/tree/59f5f77e3ddbac3ed9db93ec2cfe99ed6c5d121d). ## Set up Dataset Please follow instructions of [Instant-NVR](https://github.com/zju3dv/instant-nvr/blob/master/docs/install.md#set-up-datasets) to download ZJU-Mocap-Refine and MonoCap dataset. ## Download SMPL Models Register and download SMPL models [here](https://smplify.is.tue.mpg.de/download.php). Put the downloaded models in the folder smpl_models. Only the neutral one is needed. The folder structure should look like ``` ./ ├── ... └── assets/ ├── SMPL_NEUTRAL.pkl ``` ## :train: Training ### Training command on ZJU_MoCap_refine dataset ```bash bash run_zju_mocap_refine.sh ``` ### Training command on MonoCap dataset ```bash bash run_monocap.sh ``` ## :running_woman: Evaluation ### Evaluation command on ZJU_MoCap_refine dataset ```bash bash eval_zju_mocap_refine.sh ``` ### Evaluation command on MonoCap dataset ```bash bash eval_monocap.sh ``` ## :love_you_gesture: Citation If you find the codes of this work or the associated ReSynth dataset helpful to your research, please consider citing: ```bibtex @article{hu2023gauhuman, title={GauHuman: Articulated Gaussian Splatting from Monocular Human Videos}, author={Hu, Shoukang and Liu, Ziwei}, journal={arXiv preprint arXiv:}, year={2023} } ``` ## :newspaper_roll: License Distributed under the S-Lab License. See `LICENSE` for more information. ## :raised_hands: Acknowledgements This project is built on source codes shared by [Gaussian-Splatting](https://github.com/graphdeco-inria/gaussian-splatting), [HumanNeRF](https://github.com/chungyiweng/humannerf) and [Animatable NeRF](https://github.com/zju3dv/animatable_nerf).