chenxuanhong
32 lines
1.0 KiB
Markdown
32 lines
1.0 KiB
Markdown
# 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). |