FaceShifter
Implementation of https://arxiv.org/abs/1912.13457
Reference:
https://github.com/taotaonice/FaceShifter
Requirements
CUDA
Packages
pip install -r requirements.txt
APEX
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
Data
python face_modules/preprocess_images.py
Train
python train_aei.py
Monitor visdom
by visiting
https://HOST:8097
Demo
- Download pretrained (45 epochs)
G_latest.pth
andD_latest.pth
into./saved_models
folder - Run
python demo_cam.py
- (optional) Change
./source.jpg
to change demo source image
Todo
- Add HEAR-Net training and demo