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: