Taeu / Style-Your-Hair

Official Pytorch implementation of "Style Your Hair: Latent Optimization for Pose-Invariant Hairstyle Transfer via Local-Style-Aware Hair Alignment (ECCV 2022)"
295 stars 40 forks source link

when I inference but there is a error #8

Open kelisiya opened 2 years ago

kelisiya commented 2 years ago

TypeError: list indices must be integers or slices, not tuple

my inference code : CUDA_VISIBLE_DEVICES=3 python main.py --input_dir ./ffhq_image/ --im_path1 00016.png --im_path2 00018.png \ --output_dir ./style_your_hair_output/ \ --warp_loss_with_prev_list delta_w style_hair_slic_large \ --save_all --version final --flip_check

mrsaleh commented 1 year ago

Can you provide more information? check the files and paths exist. and Also I guess you must remove \ if you don't type the command in multiple lines.

ArsalanAli915 commented 11 months ago

@myoons I have been using T4 GPU with python=3.7.9 enviroment in anaconda. I am getting around 17 mins on inference time. Could help me know How can I reduce the time?