• Stars
    star
    115
  • Rank 305,916 (Top 7 %)
  • Language
    Jupyter Notebook
  • Created over 4 years ago
  • Updated almost 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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