Go to file
Somdev Sangwan 251fde9596
Merge pull request #91 from henryruhs/main
Introduce pre_check() and limit_resources()
2023-05-31 02:27:34 +05:30
.github/ISSUE_TEMPLATE Update issue templates 2023-05-30 18:12:38 +05:30
core Resolve absolute model path 2023-05-30 21:52:39 +02:00
demo.gif made the gif loopable 2023-05-28 20:28:58 +05:30
gui-demo.png new gui-demo 2023-05-30 04:36:04 +05:30
LICENSE Initial commit 2023-05-28 20:07:54 +05:30
README.md Normalize cli args 2023-05-30 20:43:03 +02:00
requirements.txt update pillow version 2023-05-30 19:19:11 +05:30
run.py Need to return at least 2 cores for default 2023-05-30 22:44:48 +02:00

Take a video and replace the face in it with a face of your choice. You only need one image of the desired face. No dataset, no training.

That's it, that's the software. You can watch some demos here.

demo-gif

How do I install it?

Note: The instructions may or may not work for you. Use google or look through issues people have created here to solve your problems.

There are two types of installations: basic and gpu-powered.

  • Basic: It is more likely to work on your computer but it will also be very slow. You can follow instructions for the basic install here.

  • GPU: If you have a good GPU and are ready for solving any software issues you may face, you can enable GPU which is wayyy faster. To do this, first follow the basic install instructions given above and then follow GPU-specific instructions here.

How do I use it?

Note: When you run this program for the first time, it will download some models ~300MB in size.

Executing python run.py command will launch this window: gui-demo

Choose a face (image with desired face) and the target image/video (image/video in which you want to replace the face) and click on Start. Open file explorer and navigate to the directory you select your output to be in. You will find a directory named <video_title> where you can see the frames being swapped in realtime. Once the processing is done, it will create the output file. That's it.

Don't touch the FPS checkbox unless you know what you are doing.

Additional command line arguments are given below:

options:
  -h, --help            show this help message and exit
  -f SOURCE_IMG, --face SOURCE_IMG
                        use this face
  -t TARGET_PATH, --target TARGET_PATH
                        replace this face
  -o OUTPUT_FILE, --output OUTPUT_FILE
                        save output to this file
  --gpu                 use gpu
  --keep-fps            maintain original fps
  --keep-frames         keep frames directory
  --max-memory MAX_MEMORY
                        set max memory
  --max-cores CORES_COUNT
                        set max cpu cores

Looking for a CLI mode? Using the -f/--face argument will make the program in cli mode.

Future plans

  • Improve the quality of faces in results
  • Replace a selective face throughout the video
  • Support for replacing multiple faces

Disclaimer

Deepfake software already exist. This is just an experiment to make the existing techniques better. Users are expected to use this to learn about AI and not use it for illicit or unethical purposes. Users must get consent from the concerned people before using their face and must not hide the fact that it is a deepfake when posting content online. I am not responsible for any malicious activity done through this software, this is a purely educational project aimed at exploring AI.

Credits

  • ffmpeg: for making video related operations easy
  • deepinsight: for their insightface project which provided a well-made library and models.
  • and all developers behind libraries used in this project.