From 5ec04aedc8a01139623a9c87512bf0cfa044c2f4 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Mon, 24 Aug 2020 15:00:14 -0600
Subject: [PATCH] Let noise be configurable

LQ noise is not currently configurable for some reason..
---
 codes/data/LQGT_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py
index ca872e2e..62918b4a 100644
--- a/codes/data/LQGT_dataset.py
+++ b/codes/data/LQGT_dataset.py
@@ -218,7 +218,7 @@ class LQGTDataset(data.Dataset):
         if img_GAN is not None:
             img_GAN = torch.from_numpy(np.ascontiguousarray(np.transpose(img_GAN, (2, 0, 1)))).float()
 
-        lq_noise = torch.randn_like(img_LQ) * 5 / 255
+        lq_noise = torch.randn_like(img_LQ) * self.opt['lq_noise'] / 255
         img_LQ += lq_noise
 
         if LQ_path is None: