just a little project for fast face swapping using one picture. Now supports multigpu! (Almost, check the ending of readme)

-python 3.10

-cuda 11.7 with cudnn

-everything else is installed automatically!

python main.py by default starts face swapping from camera.

flags implemented:

-f, --face: uses the face from the following image. Default: face.jpg

-t, --target: replaces the face in the following video/image. Use 0 for camera. Default: 0

-o, --output: path to output the video. Default: video.mp4

-cam-fix, --camera-fix: fix for logitech cameras that start for 40 seconds in default mode.

-res, --resolution: camera resolution, given in format WxH (ex 1920x1080). Is set for camera mode only. Default: 1920x1080

--threads: amount of threads to run the program in. Default: 2

--image: if the target is image, you have to use this flag.

--cli: run in cli mode, turns off preview and now accepts the switch of face enhancer from the command line

--face-enhancer: argument works with cli, face enhancer. In gui mode, you need to choose from gui. Available options:

1. none (default)
2. gfpgan
3. ffe (fast face enhancer)
4. codeformer
5. gfpgan_onnx
6. real_esrgan

--no-face-swapper: disables face swapper

--experimental: experimental mode to try to optimize the perfomance of reading of frames, sometimes is faster, but requires additional modules

--no-cuda: no cuda should be used (might break sometimes)

--lowmem, --low-memory: attempt to make code available for people with low VRAM, might result in lower quality

--batch: enables batch mode, after it provide a suffix, for example --batch="_test.mp4" will result in output %target%_test.mp4

--extract-output-frames: extract frames from output video. After argument write the path to folder.

--codeformer-fidelity: argument works with cli, sets up codeformer's fidelity

--blend: argument works with cli, blending amount from 0.0 to 1.0

--codeformer-skip_if_no_face: argument works with cli, Skip codeformer if no face found

--codeformer-face-upscale: argument works with cli, Upscale the face using codeformer

--codeformer-background-enhance:argument works with cli, Enhance the background using codeformer

--codeformer-upscale: argument works with cli, the amount of upscale to apply to the frame using codeformer

--select-face: change the face you want, not all faces. After the argument add the path to the image with face from the video. (Just open video in video player, screenshot the frame and save it to file. Put this filename after --select-face argument)

--optimization: choose the mode of the model: fp32 (default), fp16 (smaller, might be faster), int8 (doesnt work properly on old gpus, I dont know about new once, please test. On old gpus it uses cpu)

--fast-load: try to load as fast as possible, might break something sometimes

--bbox-adjust: adjustments to do for the box around the face: x1,y1 coords of left top corner and x2,y2 are bottom right. Give in the form x1xy1xx2xy2 (default: 50x50x50x50). Just try to play to understand

-vcam, --virtual-camera: allows to use OBS virtual camera as output source. Please install obs to make sure it works

example: python main.py -f test.jpg -t "C:/Users/user/Desktop/video.mp4" -o output/test.mp4 --threads 12

fast enhancer is still in development, color correction is needed! Sorry for inconvenience, still training the model.

To choose the gpu you want to run on: in globalsz.py, on the line with select_gpu = None you can make it select_gpu = [0, 1] or something similar (these numbers are id of gpus, starting from 0).

to use all gpus, select_gpu = None

Multigpu mode for now only supports just face swapping, without the enhancer!!! So if you want enhancer to work, for now select only one gpu.