Skip to content

Latest commit

 

History

History
57 lines (45 loc) · 2.73 KB

usage.md

File metadata and controls

57 lines (45 loc) · 2.73 KB

Usage

Before running, please make sure you have installed the environment and downloaded requested files according to the preparation guidance.

Simple face swapping for already face-aligned images

python test_one_image.py --isTrain false  --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path crop_224/6.jpg --pic_b_path crop_224/ds.jpg --output_path output/

Face swapping for video

  • Swap only one face within the video(the one with highest confidence by face detection).
python test_video_swapsingle.py --isTrain false  --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path ./demo_file/Iron_man.jpg --video_path ./demo_file/mutil_people_1080p.mp4 --output_path ./output/mutil_test_swapsingle.mp4 --temp_path ./temp_results
  • Swap all faces within the video.
python test_video_swapmutil.py --isTrain false  --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path ./demo_file/Iron_man.jpg --video_path ./demo_file/mutil_people_1080p.mp4 --output_path ./output/mutil_test_swapmutil.mp4 --temp_path ./temp_results

Face swapping for Arbitrary images

  • Swap only one face within one image(the one with highest confidence by face detection). The result would be saved to ./output/result_whole_swapsingle.jpg
python test_wholeimage_swapsingle.py --isTrain false  --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path ./demo_file/Iron_man.jpg --pic_b_path ./demo_file/mutil_people.jpg --output_path ./output/
  • Swap all faces within one image. The result would be saved to ./output/result_whole_swapmutil.jpg
python test_wholeimage_swapmutil.py --isTrain false  --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path ./demo_file/Iron_man.jpg --pic_b_path ./demo_file/mutil_people.jpg --output_path ./output/

Difference between single face swapping and all face swapping are shown below.

Parameters

Parameters Function
--name The SimSwap training logs name
--pic_a_path Path of image with the target face
--pic_b_path Path of image with the source face to swap
--video_path Path of video with the source face to swap
--temp_path Path to store intermediate files
--output_path Path of directory to store the face swapping result

Note

We expect users to have GPU with at least 8G memory. For those who do not, we will provide Colab Notebook implementation in the future.