This repository contains the official implementation for Towards Unified 3D Hair Reconstruction from Single-View Portraits.

• If you would like to use our data, please contact paul.yj.zheng@gmail.com
• 2024.11.24: release the code!

git clone --recursive https://github.com/PAULYZHENG/UniHair.git
cd UniHair

# change CUDA to 11.3 before creating the env
conda env create -f environment.yml
conda activate unihair

sh setup.sh

Tested on Ubuntu 22 with torch 1.12.1 & CUDA 11.3 on 3090/4090.

Put images you want to test in data/img/

sh scripts/run_hairalign.sh
sh scripts/run_unihair.sh

Find your results in data/logs/

This repository is based on some excellent works; many thanks to all the authors!

• DreamGaussian
• gaussian-splatting and diff-gaussian-rasterization
• zero123 and zero123-hf
• 3DFFA_V2
• SAM

@inproceedings{zheng2024towards,
  title={Towards Unified 3D Hair Reconstruction from Single-View Portraits},
  author={Zheng, Yujian and Qiu, Yuda and Jin, Leyang and Ma, Chongyang and Huang, Haibin and Zhang, Di and Wan, Pengfei and Han, Xiaoguang},
  booktitle={SIGGRAPH Asia 2024 Conference Papers},
  pages={1--11},
  year={2024}
}