The official pytorch implementation of ATW paper, Animating Through Warping: an efficient Method for High-Quality Facial Expression Animation
- Python 3
- PyTorch 0.4.1
- visdom (optional, only for training with browser visualizer)
- imageio (optional, only for generating GIF image in testing)
- Clone this repo:
git clone ssh://[email protected]:2222/HiLens/hicam/facial_animation.git
cd facial_animation
conda create --name <env_name> --file requirements.txt
- prepare the training datasets
- uncomment the corresponding items in run.sh, then run the following command:
bash run.sh
- download the pretrained models from google drive pretrained models
- extract the file and put the folder pretrained_ckpt/ under ./ATW/
- uncomment the corresponding items in the run.sh, then run the following command:
bash run.sh
Feel free to contact me if you need any help from me related to this project.