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

Getting error on inference when using reference image hairstyle to paste on input image. #45

Open ArsalanAli915 opened 1 year ago

ArsalanAli915 commented 1 year ago

e1 I am getting error when I try to take inference. I am using this command python scripts/inference.py \ --exp_dir=/content/resultss \ --editing_type=both \ --input_type=image_image \ --hairstyle_ref_img_test_path=/content/oriental1.png \ --color_ref_img_test_path=/content/oriental1.png \ --num_of_ref_img 1 \ --checkpoint_path=/content/drive/MyDrive/data/hairclip.pt \ --latents_test_path=/content/drive/MyDrive/data/latents.pt \ What I am trying to is to take transfer hairstyle of refrence Image to input Image. I have converted input image to e4e to get latent code. Please do let me know. thnx.