GauHuman: Articulated Gaussian Splatting from Monocular Human Videos
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).