update

.gitignore

@@ -125,4 +125,5 @@ env/
 wandb/
 train_logs/
 test_logs/
-arcface_ckpt/
+arcface_ckpt/
+GUI/

README.md

@@ -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```.
 

utilities/transfer_checkpoint.py

@@ -5,7 +5,7 @@
 # Created Date: Wednesday February 3rd 2021
 # Author: Chen Xuanhong
 # 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
 # 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 init as init
 import os
-import numpy as np
-import scipy.io as io
+
 
 class RepSRPlain_pixel(nn.Module):
     """Networks consisting of Residual in Residual Dense Block, which is used