From 736d45ec00843054fe1135878a2da6c1df8435e9 Mon Sep 17 00:00:00 2001
From: chenxuanhong <chenxuanhongzju@outlook.com>
Date: Mon, 17 Jan 2022 13:26:59 +0800
Subject: [PATCH] update

---
 .gitignore                       |  3 ++-
 README.md                        | 15 +++++++++++++--
 utilities/transfer_checkpoint.py |  5 ++---
 3 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/.gitignore b/.gitignore
index 8a3da5d..3e65a47 100644
--- a/.gitignore
+++ b/.gitignore
@@ -125,4 +125,5 @@ env/
 wandb/
 train_logs/
 test_logs/
-arcface_ckpt/
\ No newline at end of file
+arcface_ckpt/
+GUI/
\ No newline at end of file
diff --git a/README.md b/README.md
index 5a36251..eb238b5 100644
--- a/README.md
+++ b/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```.
 
diff --git a/utilities/transfer_checkpoint.py b/utilities/transfer_checkpoint.py
index 207e842..a6c3dd9 100644
--- a/utilities/transfer_checkpoint.py
+++ b/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