michaildoukas / head2head

PyTorch implementation for Head2Head and Head2Head++. It can be used to fully transfer the head pose, facial expression and eye movements from a source video to a target identity.
MIT License
309 stars 44 forks source link

Fix bug of 'scripts/download_dataset.py' #14

Open BbChip0103 opened 3 years ago

BbChip0103 commented 3 years ago

Hi, thank you for your work. When I tried to download dataset using 'scripts/download_dataset.py', some error was occurred. As I read the code, the code download dataset and unzip in some directory, the bug can be occurred when the directory doesn't exist. So I add some line that make directory before download the data.

Thank you.