Simswap++

Dependencies

• python >= 3.7
• yaml (pip install pyyaml)
• paramiko (For ssh file transportation)
• pytorch >= 1.9
• pillow
• torchvision
• opencv
• matplotlib
• timm
• cupy (for involution) you need to create a new env in anaconda (conda install pytorch==1.10.1 cudatoolkit==10.2.89 cupy==10.1.0 -c pytorch -c conda-forge)

logger

• wandb (pip install wandb)

OR

• wandb (pip install wandb)

Logger is an option setting, which can be adjust with train.py --logger [wandb, tensorbaord, None]

Usage

• To configure the project in the main.py.

Acknowledgement

Related Projects

Learn about our other projects [RainNet], [Sketch Generation], [CooGAN], [Knowledge Style Transfer], [Youtube downloader].