From a25ee9464d88360ffc3475f3ffb97880ee18f337 Mon Sep 17 00:00:00 2001
From: XintaoWang <wxt1994@126.com>
Date: Sun, 1 Sep 2019 22:20:10 +0800
Subject: [PATCH] test w/o GT

---
 codes/data/LQ_dataset.py                |  3 ++-
 codes/options/test/test_ESRGAN_woGT.yml | 26 +++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 codes/options/test/test_ESRGAN_woGT.yml

diff --git a/codes/data/LQ_dataset.py b/codes/data/LQ_dataset.py
index e7ad7dd3..277c1671 100644
--- a/codes/data/LQ_dataset.py
+++ b/codes/data/LQ_dataset.py
@@ -11,6 +11,7 @@ class LQDataset(data.Dataset):
     def __init__(self, opt):
         super(LQDataset, self).__init__()
         self.opt = opt
+        self.data_type = self.opt['data_type']
         self.paths_LQ, self.paths_GT = None, None
         self.LQ_env = None  # environment for lmdb
 
@@ -27,7 +28,7 @@ class LQDataset(data.Dataset):
         LQ_path = None
 
         # get LQ image
-        LQ_path = self.LQ_path[index]
+        LQ_path = self.paths_LQ[index]
         resolution = [int(s) for s in self.sizes_LQ[index].split('_')
                       ] if self.data_type == 'lmdb' else None
         img_LQ = util.read_img(self.LQ_env, LQ_path, resolution)
diff --git a/codes/options/test/test_ESRGAN_woGT.yml b/codes/options/test/test_ESRGAN_woGT.yml
new file mode 100644
index 00000000..24ab5b62
--- /dev/null
+++ b/codes/options/test/test_ESRGAN_woGT.yml
@@ -0,0 +1,26 @@
+name: RRDB_ESRGAN_x4
+suffix: ~  # add suffix to saved images
+model: sr
+distortion: sr
+scale: 4
+crop_border: ~  # crop border when evaluation. If None(~), crop the scale pixels
+gpu_ids: [0]
+
+datasets:
+  test_1:  # the 1st test dataset
+    name: set14
+    mode: LQ
+    dataroot_LQ: ../datasets/val_set14/Set14_bicLRx4
+
+#### network structures
+network_G:
+  which_model_G: RRDBNet
+  in_nc: 3
+  out_nc: 3
+  nf: 64
+  nb: 23
+  upscale: 4
+
+#### path
+path:
+  pretrain_model_G: ../experiments/pretrained_models/RRDB_ESRGAN_x4.pth