CalvinBackup/SimSwapPlus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+1
View File
@@ -126,3 +126,4 @@ wandb/
train_logs/ train_logs/
test_logs/ test_logs/
arcface_ckpt/ arcface_ckpt/
GUI/
README.md
+13 -2
View File
@@ -1,17 +1,28 @@
# Project Name # Simswap++
## Dependencies ## Dependencies
- python - python
- yaml (pip install pyyaml) - yaml (pip install pyyaml)
- paramiko (For ssh file transportation) - paramiko (For ssh file transportation)
- pytorch - pytorch > 1.8
- tkinter (For GUI) - tkinter (For GUI)
- pillow - pillow
- torchvision - torchvision
- opencv - opencv
- matplotlib
- timm
## logger
- tensorboard (pip install tensorboard) - tensorboard (pip install tensorboard)
- tensorboardX (pip install tensorboardX) - 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 ## Usage
- To configure the project in the ```main.py```. - To configure the project in the ```main.py```.
utilities/transfer_checkpoint.py
+2 -3
View File
@@ -5,7 +5,7 @@
# Created Date: Wednesday February 3rd 2021 # Created Date: Wednesday February 3rd 2021
# Author: Chen Xuanhong # Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com # Email: chenxuanhongzju@outlook.com
# Last Modified: Thursday, 4th February 2021 1:27:09 pm # Last Modified: Monday, 17th January 2022 1:25:56 pm
# Modified By: Chen Xuanhong # Modified By: Chen Xuanhong
# Copyright (c) 2021 Shanghai Jiao Tong University # Copyright (c) 2021 Shanghai Jiao Tong University
############################################################# #############################################################
@@ -14,8 +14,7 @@ from torch import nn as nn
from torch.nn import functional as F from torch.nn import functional as F
from torch.nn import init as init from torch.nn import init as init
import os import os
import numpy as np
import scipy.io as io
class RepSRPlain_pixel(nn.Module): class RepSRPlain_pixel(nn.Module):
"""Networks consisting of Residual in Residual Dense Block, which is used """Networks consisting of Residual in Residual Dense Block, which is used