wty-ustc / HairCLIP

[CVPR 2022] HairCLIP: Design Your Hair by Text and Reference Image
GNU Lesser General Public License v2.1
541 stars 68 forks source link

Hairstyles can only show so much #14

Closed ZziTaiLeo closed 2 years ago

ZziTaiLeo commented 2 years ago

First of all , thanks for you excellent work! There are many hairstyles in hairstyle.txt, but actually I found only a few styles in result images after trying all styles. More or less repeat the following images.

the following is my command:

python scripts/inference.py 
--exp_dir=../result/test_1/
--checkpoint_path=../pretrained_models/hairclip.pt
--latents_test_path=../inference_data/test_1/latent.pt
--editing_type=hairstyle
--input_type=text
--hairstyle_description="hairstyle_list.txt"

What's the problem? Should I train with my own dataset?

I list some hairstyles which have the same effect:

wty-ustc commented 2 years ago
  1. Some of the text hairstyle descriptions are very similar.
  2. As stated in the limitations section of our paper, no reasonable results can be given for descriptions that are not in the domains of StyleGAN. For example, some hairstyles are only available for ladies.
  3. You can reduce the number of hairstyle text descriptions to retrain HairCLIP, the training process may take less than 1 day, and this approach may be effective.