# 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***

- Do not need to install tensorboard and tensorboardX any more.

***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]](https://neuralchen.github.io/RainNet), [[Sketch Generation]](https://github.com/TZYSJTU/Sketch-Generation-with-Drawing-Process-Guided-by-Vector-Flow-and-Grayscale), [[CooGAN]](https://github.com/neuralchen/CooGAN), [[Knowledge Style Transfer]](https://github.com/AceSix/Knowledge_Transfer), [[Youtube downloader]](https://github.com/AIARTSJTU/YoutubeDataCollector).