zomin28 / free_manipulation_system Goto Github PK

Creating a conda environment named pytorch：

conda create -n pytorch python=3.7

Install the required dependency packages. Given in requirements.txt.

StarGANv2.

bash download.sh celeba-hq-dataset
bash download.sh pretrained-network-celeba-hq
bash download.sh wing

HiSD. Put it in Free_manipulation_system\HiSD\checkpoints

HFGI. Put it in Free_manipulation_system\HFGI\checkpoint

SimSwap. Put them in DF-RAP/SimSwap/arcface_model/ and DF-RAP/SimSwap/checkpoints/ .

running run.bat.

If you need to call it in your script, you can use the following：

from manipulation import manipulate
manipulate(original_path, algorithm, dev, reference_path, reference)

https://github.com/clovaai/stargan-v2

https://github.com/imlixinyang/HiSD

https://github.com/Tengfei-Wang/HFGI

https://github.com/neuralchen/SimSwap