Environment configuration

The codes are based on python3.8+, CUDA version 11.0+. The specific configuration steps are as follows:

1. Create conda environment

   conda create -n fadm python=3.8
   conda activate fadm

2. Install pytorch

   conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

3. Installation profile

   pip install -r requirements.txt

Pre-trained checkpoint

Checkpoints can be found under the following link: link.

• Download the data.zip, and unzip to the path modules/deca.
• Download the FOMM.pth, Facevid.pth, and diffusion.pth to the path modules/FOMM/ckpt/, modules/Facevid/ckpt/, and modules/diffusion/ckpt/ respectively.

Demo testing

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