Skip to content

Heonozis/FaceShifter-pytorch

Repository files navigation

FaceShifter

Implementation of https://arxiv.org/abs/1912.13457

Reference:

https://github.com/taotaonice/FaceShifter

Requirements

CUDA

Cuda 10.1

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

  1. Get embedding model weights

  2. Get CelebA dataset and extract to ../img_align_celeba

  3. Preprocess data

python face_modules/preprocess_images.py

Train

python train_aei.py

Monitor visdom by visiting

https://HOST:8097

Demo

  1. Download pretrained (45 epochs) G_latest.pth and D_latest.pth into ./saved_models folder
  2. Run python demo_cam.py
  3. (optional) Change ./source.jpg to change demo source image

Todo

  • Add HEAR-Net training and demo

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published