chenxuanhong
1.0 KiB
1.0 KiB
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], [Sketch Generation], [CooGAN], [Knowledge Style Transfer], [Youtube downloader].