README.md

StableHair v2

Stable-Hair v2: Real-World Hair Transfer via Multiple-View Diffusion Model
Kuiyuan Sun*, Yuxuan Zhang*, Jichao Zhang*, Jiaming Liu, Wei Wang, Nicu Sebe, Yao Zhao
*Equal Contribution
Beijing Jiaotong University, Shanghai Jiaotong University, Ocean University of China, Tiamat AI, University of Trento
Arxiv, Project

Bald | Reference | Multiple View | Original Video
Bald | Reference | Multiple View | Original Video

Environments

conda create -n stablehairv2 python=3.10

pip install -r requirements.txt

Results

Pretrained Model

Name	Model
motion_module-41400000.pth	:link:
pytorch_model_1.bin	:link:
pytorch_model_2.bin	:link:
pytorch_model_3.bin	:link:
pytorch_model.bin	:link:

Multiple View Hair Transfer

Please use ``gdown''' to download the pretrained model and save it in your model_path

python test_stablehairv2.py --pretrained_model_name_or_path "stable-diffusion-v1-5/stable-diffusion-v1-5" \
       --image_encoder "openai/clip-vit-large-patch14" --output_dir [Your_output_dir]  \
       --num_validation_images 1 --validation_ids ./test_imgs/bald.jpg \
       --validation_hairs ./test_imgs/ref1.jpg --model_path [Your_model_path]

Our V1 version

StableHair v2 is an improved version of StableHair (AAAI 2025)