This commit is contained in:
chenxuanhong
2022-01-17 13:26:59 +08:00
parent 036251d36e
commit 736d45ec00
3 changed files with 17 additions and 6 deletions
+13 -2
View File
@@ -1,17 +1,28 @@
# Project Name
# Simswap++
## Dependencies
- python
- yaml (pip install pyyaml)
- paramiko (For ssh file transportation)
- pytorch
- pytorch > 1.8
- tkinter (For GUI)
- pillow
- torchvision
- opencv
- matplotlib
- timm
## logger
- tensorboard (pip install tensorboard)
- tensorboardX (pip install tensorboardX)
***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```.