Need help with faceswap-GAN?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

shaoanlu
3.0K Stars 782 Forks 277 Commits 93 Opened issues

Description

A denoising autoencoder + adversarial losses and attention mechanisms for face swapping.

Services available

!
?

Need anything else?

Contributors list

# 7,492
Jupyter...
Python
gans
face-ve...
263 commits
# 165,198
Haskell
Shell
PureScr...
Jupyter...
1 commit
# 48,611
JavaScr...
babel-p...
Babel
face-sw...
1 commit
# 169,687
Jupyter...
Python
face-sw...
Shell
1 commit

faceswap-GAN

Adding Adversarial loss and perceptual loss (VGGface) to deepfakes'(reddit user) auto-encoder architecture.

Updates

| Date    | Update | | ------------- | ------------- |
| 2018-08-27     | Colab support: A colab notebook for faceswap-GAN v2.2 is provided.| | 2018-07-25     | Data preparation: Add a new notebook for video pre-processing in which MTCNN is used for face detection as well as face alignment.| | 2018-06-29     | Model architecture: faceswap-GAN v2.2 now supports different output resolutions: 64x64, 128x128, and 256x256. Default

RESOLUTION = 64
can be changed in the config cell of v2.2 notebook.| | 2018-06-25     | New version: faceswap-GAN v2.2 has been released. The main improvements of v2.2 model are its capability of generating realistic and consistent eye movements (results are shown below, or Ctrl+F for eyes), as well as higher video quality with face alignment.| | 2018-06-06     | Model architecture: Add a self-attention mechanism proposed in SAGAN into V2 GAN model. (Note: There is still no official code release for SAGAN, the implementation in this repo. could be wrong. We'll keep an eye on it.)|

Google Colab support

Here is a playground notebook for faceswap-GAN v2.2 on Google Colab. Users can train their own model in the browser.

[Update 2019/10/04] There seems to be import errors in the latest Colab environment due to inconsistent version of packages. Please make sure that the Keras and TensorFlow follow the version number shown in the requirement section below.

Descriptions

faceswap-GAN v2.2

  • FaceSwapGANv2.2traintest.ipynb

    • Notebook for model training of faceswap-GAN model version 2.2.
    • This notebook also provides code for still image transformation at the bottom.
    • Require additional training images generated through prepbinarymasks.ipynb.
  • FaceSwapGANv2.2videoconversion.ipynb

    • Notebook for video conversion of faceswap-GAN model version 2.2.
    • Face alignment using 5-points landmarks is introduced to video conversion.
  • prepbinarymasks.ipynb

    • Notebook for training data preprocessing. Output binary masks are save in
      ./binary_masks/faceA_eyes
      and
      ./binary_masks/faceB_eyes
      folders.
    • Require face_alignment package. (An alternative method for generating binary masks (not requiring
      face_alignment
      and
      dlib
      packages) can be found in MTCNNvideofacedetectionalignment.ipynb.)
  • MTCNNvideofacedetectionalignment.ipynb

    • This notebook performs face detection/alignment on the input video.
    • Detected faces are saved in
      ./faces/raw_faces
      and
      ./faces/aligned_faces
      for non-aligned/aligned results respectively.
    • Crude eyes binary masks are also generated and saved in
      ./faces/binary_masks_eyes
      . These binary masks can serve as a suboptimal alternative to masks generated through prepbinarymasks.ipynb.

Usage 1. Run MTCNNvideofacedetectionalignment.ipynb to extract faces from videos. Manually move/rename the aligned face images into

./faceA/
or
./faceB/
folders. 2. Run prepbinarymasks.ipynb to generate binary masks of training images. - You can skip this pre-processing step by (1) setting
use_bm_eyes=False
in the config cell of the train_test notebook, or (2) use low-quality binary masks generated in step 1. 3. Run FaceSwapGANv2.2traintest.ipynb to train models. 4. Run FaceSwapGANv2.2videoconversion.ipynb to create videos using the trained models in step 3.

Miscellaneous

Training data format

  • Face images are supposed to be in
    ./faceA/
    or
    ./faceB/
    folder for each taeget respectively.
  • Images will be resized to 256x256 during training.

Generative adversarial networks for face swapping

1. Architecture

enc_arch3d

dec_arch3d

dis_arch3d

2. Results

  • Improved output quality: Adversarial loss improves reconstruction quality of generated images. trump_cage

  • Additional results: This image shows 160 random results generated by v2 GAN with self-attention mechanism (image format: source -> mask -> transformed).

  • Evaluations: Evaluations of the output quality on Trump/Cage dataset can be found here.

The Trump/Cage images are obtained from the reddit user deepfakes' project on pastebin.com.

3. Features

  • VGGFace perceptual loss: Perceptual loss improves direction of eyeballs to be more realistic and consistent with input face. It also smoothes out artifacts in the segmentation mask, resulting higher output quality.

  • Attention mask: Model predicts an attention mask that helps on handling occlusion, eliminating artifacts, and producing natrual skin tone.

  • Configurable input/output resolution (v2.2): The model supports 64x64, 128x128, and 256x256 outupt resolutions.

  • Face tracking/alignment using MTCNN and Kalman filter in video conversion:

    • MTCNN is introduced for more stable detections and reliable face alignment (FA).
    • Kalman filter smoothen the bounding box positions over frames and eliminate jitter on the swapped face. comp_FA
  • Eyes-aware training: Introduce high reconstruction loss and edge loss in eyes area, which guides the model to generate realistic eyes.

Frequently asked questions and troubleshooting

1. How does it work?

  • The following illustration shows a very high-level and abstract (but not exactly the same) flowchart of the denoising autoencoder algorithm. The objective functions look like this. flow_chart #### 2. Previews look good, but it does not transform to the output videos?
  • Model performs its full potential when the input images are preprocessed with face alignment methods.
    • readme_note001

Requirements

Acknowledgments

Code borrows from tjwei, eriklindernoren, fchollet, keras-contrib and reddit user deepfakes' project. The generative network is adopted from CycleGAN. Weights and scripts of MTCNN are from FaceNet. Illustrations are from irasutoya.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.