facebookresearch / goliath

Goliath Dataset and Official PyTorch Implementation of RelightableHands, Relightable Gaussian Codec Avatars, and Driving-Signal Aware Full-Body Avatars.
Other
133 stars 9 forks source link

Goliath

Together with Ava-256, part of Codec Avatar Studio

We provide 4 sets of captures; for each subject:

And code to train personalized decoders:

goliath

Please refer to the samples to get a sense of what the data looks like.

Disclaimer

This is a pre-release.

Dependencies

See requirements.txt/environment.yaml

Repository structure

Data

Besides camera, views, we also provide segmentations, 3d keypoints, registered and unregistered meshes, as well as light information when available.

https://github.com/facebookresearch/goliath/assets/3733964/3052d8ee-e2b8-48e0-9715-9f0c2e6f6e72

Access to the dataset is currently gated. Please email julietamartinez@meta.com, preferrably from an institutional email, to get access to the data.

Compiling and installing extensions

cd extensions/{mvpraymarch,sgutils,utils}
make

Training

(You may have to add the directory of the codebase to your PYTHONPATH)

python ca_code/scripts/run_train.py <config.yml>

or simply

python -m ca_code.scripts.run_train <config.yml>

Visualization (Relighting)

python ca_code/scripts/run_vis_relight.py <config.yml>

Evaluation

TODO:

License

See LICENSE.

Citation

If you use this repository, please cite relevant paper(s).

Full-body Avatars

@article{bagautdinov2021driving,
  title={Driving-signal aware full-body avatars},
  author={Bagautdinov, Timur and Wu, Chenglei and Simon, Tomas and Prada, Fabi{\'a}n and Shiratori, Takaaki and Wei, Shih-En and Xu, Weipeng and Sheikh, Yaser and Saragih, Jason},
  journal={ACM Transactions on Graphics (TOG)},
  volume={40},
  number={4},
  pages={1--17},
  year={2021},
  publisher={ACM New York, NY, USA}
}

Relightable Head Avatars

@inproceedings{saito2024rgca,
  author = {Shunsuke Saito and Gabriel Schwartz and Tomas Simon and Junxuan Li and Giljoo Nam},
  title = {Relightable Gaussian Codec Avatars},
  booktitle = {CVPR},
  year = {2024},
}

Relightable Hand Avatars

@inproceedings{iwase2023relightablehands,
  title={Relightablehands: Efficient neural relighting of articulated hand models},
  author={Iwase, Shun and Saito, Shunsuke and Simon, Tomas and Lombardi, Stephen and Bagautdinov, Timur and Joshi, Rohan and Prada, Fabian and Shiratori, Takaaki and Sheikh, Yaser and Saragih, Jason},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={16663--16673},
  year={2023}
}