Create conda environment
conda create -n fadm python=3.8
conda activate fadm
Install pytorch
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
Installation profile
pip install -r requirements.txt
Checkpoints can be found under the following link: link.
data.zip
, and unzip to the path modules/deca
.FOMM.pth
, Facevid.pth
, and diffusion.pth
to the path modules/FOMM/ckpt/
, modules/Facevid/ckpt/
, and modules/diffusion/ckpt/
respectively.To run a reenactment demo, download checkpoint and run the following command:
python demo.py
And run the following command for training:
sh run.sh