wty-ustc / HairCLIP

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

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

Open ArsalanAli915 opened 8 months ago

ArsalanAli915 commented 8 months 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.