Mastrom10 / faceswap

Faceswap project based on original /r/Deepfakes thread. Many thanks to him!

Home Page:https://www.reddit.com/user/deepfakes/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This is the code from deepfakes' faceswap project.


Important Notice this is NOT the repo of https://reddit.com/user/deepfakes . This is a repo/user created on purpose for this project. (If deepfakes want to take ownership of this repo he is welcomed to do so. (Simply raise an issue))

deepfakes_faceswap

You'll find here the code for the project. The original code is in the first commits of the master. Follow new features here: https://github.com/deepfakes/faceswap/projects/1

How to contribute

For devs

  • Read this README entirely
  • Fork the repo
  • Download the data with the link provided below
  • Play with it
  • Check issues with the 'dev' tag

For non-dev advanced users

  • Read this README entirely
  • Clone the repo
  • Download the data with the link provided below
  • Play with it
  • Check issues with the 'advuser' tag

For end-users

Sadly the project is not yet ready to use for non technical people, it will improve over time, so check for updates regularly and discuss with other. What you can do for now is collect the photos you want to see processed and shere them with others. Submit them with a comment here

For haters

Sorry no time for that

Overview

The project has multiple entry points. You will have to:

  • Gather photos (or use the one provided in the training data provided below)
  • Extract faces from your raw photos
  • Train a model on your photos (or use the one provided in the training data provided below)
  • Convert your sources with the model

Extract

From your setup folder, run python extract.py. This will take photos from src folder and extract faces into extract folder.

Train

From your setup folder, run python train.py. This will take photos from data/trump and data/cage folder and train a model that will be saved inside the models folder.

Convert

From your setup folder, run python convert_photo.py. This will take photos from original folder and apply new faces into modified folder.

Note: there is no conversion for video yet. You can use MJPG to convert video into photos,, process images, and convert images back to video

Training Data

Whole project with training images and trained model (~300MB):
https://anonfile.com/p7w3m0d5be/face-swap.zip or click here to download

How To setup and run the project

Setup

Clone the repo and setup you environment. There is a Dockerfile that should kickstart you. Otherwise you can setup things manually, see in the Dockerfiles for dependencies

Main Requirements: Python 3 Opencv 3 Tensorflow 1.3+(?) Keras 2

You also need a modern GPU with CUDA support for best performance

Some tips:

Reuse existing models will train much faster than start from nothing.
If there are not enough training data, start with someone looks similar, then switch the data.

Docker

If you prefer using Docker, You can start the project with:

  • Build: docker build -t deepfakes .
  • Run: docker run --rm --name deepfakes -v [src_folder]:/srv -it deepfakes bash . bash can be replaced by your command line Note that the Dockerfile does not have all good requirments, so it will fail on some python 3 commands. Also note that it does not have a GUI output, so the train.py will fail on showing image. You can comment this, or save it as a file.

About

Faceswap project based on original /r/Deepfakes thread. Many thanks to him!

https://www.reddit.com/user/deepfakes/


Languages

Language:Python 100.0%