search

harlanhong / CVPR2022-DaGAN

Official code for CVPR2022 paper: Depth-Aware Generative Adversarial Network for Talking Head Video Generation
https://harlanhong.github.io/publications/dagan.html
Other
957 stars 125 forks source link

Error in running a demo version! #32

Closed muxiddin19 closed 2 years ago

muxiddin19 commented 2 years ago

Hello! Thanks for sharing openly amazing work! My research is also related to generating talking faces. I face error when tried to run: CUDA_VISIBLE_DEVICES=0 python demo.py --config config/vox-adv-256.yaml --driving_video data/2.mp4 --source_image data/2.jpg --checkpoint depth/models/weights_19/encoder.pth --relative --adapt_scale --kp_num 15 --generator DepthAwareGenerator image Can you please correct me where I made mistakes while running the demo one?

harlanhong commented 2 years ago

Could you please crop the whole message of your error? I can only find that the error occurs when loading the checkpoints.

muxiddin19 commented 2 years ago

Thanks for your reply! Actually, it was the full message. When I made some changes like: CUDA_VISIBLE_DEVICES=0 python demo.py --config config/vox-adv-256.yaml --driving_video data/2.mp4 --source_image data/2.jpg --checkpoint checkpoints/DaGAN_vox_adv_256.pth.tar --relative --adapt_scale --kp_num 15 --generator DepthAwareGenerator --result_video results/example_out.mp4 --find_best_frame It worked without any error. Here are some screenshots: image

image I think I will close the issue, as it was solved. Thank you!

harlanhong commented 2 years ago

Oh, I see. You just mistook the checkpoint of the depth network for the Dagan.