From 3a59e60b29c392cbc877eb9fbb816a2e2a33c45c Mon Sep 17 00:00:00 2001
From: pp00704831 <57279608+pp00704831@users.noreply.github.com>
Date: Sat, 16 Jul 2022 21:01:00 +0800
Subject: [PATCH] Add files via upload

---
 predict_GoPro_test_results.py      | 2 +-
 predict_HIDE_results.py            | 2 +-
 predict_RealBlur_J_test_results.py | 2 +-
 predict_RealBlur_R_test_results.py | 2 +-
 train_Stripformer_gopro.py         | 2 +-
 train_Stripformer_pretrained.py    | 2 +-
 6 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/predict_GoPro_test_results.py b/predict_GoPro_test_results.py
index 82fa930..2ed13e6 100644
--- a/predict_GoPro_test_results.py
+++ b/predict_GoPro_test_results.py
@@ -18,7 +18,7 @@ def get_args():
 if __name__ == '__main__':
     args = get_args()
     with open('config/config_Stripformer_gopro.yaml') as cfg:
-        config = yaml.load(cfg)
+        config = yaml.safe_load(cfg)
     blur_path = './datasets/GoPro/test/blur/'
     out_path = './out/Stripformer_GoPro_results'
     if not os.path.isdir(out_path):
diff --git a/predict_HIDE_results.py b/predict_HIDE_results.py
index 5e40d94..5852221 100644
--- a/predict_HIDE_results.py
+++ b/predict_HIDE_results.py
@@ -18,7 +18,7 @@ def get_args():
 if __name__ == '__main__':
     args = get_args()
     with open('config/config_Stripformer_gopro.yaml') as cfg:
-        config = yaml.load(cfg)
+        config = yaml.safe_load(cfg)
     blur_path = './datasets/HIDE/blur/'
     out_path = './out/Stripformer_HIDE_results'
     if not os.path.isdir(out_path):
diff --git a/predict_RealBlur_J_test_results.py b/predict_RealBlur_J_test_results.py
index 36d34fb..d870bdf 100644
--- a/predict_RealBlur_J_test_results.py
+++ b/predict_RealBlur_J_test_results.py
@@ -19,7 +19,7 @@ def get_args():
 if __name__ == '__main__':
     args = get_args()
     with open('config/config_Stripformer_gopro.yaml') as cfg:
-        config = yaml.load(cfg)
+        config = yaml.safe_load(cfg)
     blur_path = './datasets/Realblur_J/test/blur/'
     out_path = './out/Stripformer_realblur_J_results'
     if not os.path.isdir(out_path):
diff --git a/predict_RealBlur_R_test_results.py b/predict_RealBlur_R_test_results.py
index e889df2..4ab8df2 100644
--- a/predict_RealBlur_R_test_results.py
+++ b/predict_RealBlur_R_test_results.py
@@ -19,7 +19,7 @@ def get_args():
 if __name__ == '__main__':
     args = get_args()
     with open('config/config_Stripformer_gopro.yaml') as cfg:
-        config = yaml.load(cfg)
+        config = yaml.safe_load(cfg)
     blur_path = './datasets/Realblur_R/test/blur/'
     out_path = './out/Stripformer_realblur_R_results'
     model = get_generator(config['model'])
diff --git a/train_Stripformer_gopro.py b/train_Stripformer_gopro.py
index 2194587..08bc177 100644
--- a/train_Stripformer_gopro.py
+++ b/train_Stripformer_gopro.py
@@ -148,7 +148,7 @@ class Trainer:
 
 if __name__ == '__main__':
     with open('config/config_Stripformer_gopro.yaml', 'r') as f:
-        config = yaml.load(f)
+        config = yaml.safe_load(f)
     # setup
     torch.backends.cudnn.enabled = True
     torch.backends.cudnn.benchmark = True
diff --git a/train_Stripformer_pretrained.py b/train_Stripformer_pretrained.py
index aac7ab6..1ae3033 100644
--- a/train_Stripformer_pretrained.py
+++ b/train_Stripformer_pretrained.py
@@ -142,7 +142,7 @@ class Trainer:
 
 if __name__ == '__main__':
     with open('config/config_Stripformer_pretrained.yaml', 'r') as f:
-        config = yaml.load(f)
+        config = yaml.safe_load(f)
     # setup
     torch.backends.cudnn.enabled = True
     torch.backends.cudnn.benchmark = True