mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Let noise be configurable

Browse Source 
LQ noise is not currently configurable for some reason..
This commit is contained in:
James Betker 2020-08-24 15:00:14 -06:00
parent f9276007a8
commit 5ec04aedc8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/LQGT_dataset.py
View File

@ -218,7 +218,7 @@ class LQGTDataset(data.Dataset):
if img_GAN is not None: if img_GAN is not None:
img_GAN = torch.from_numpy(np.ascontiguousarray(np.transpose(img_GAN, (2, 0, 1)))).float() 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 img_LQ += lq_noise
if LQ_path is None: if LQ_path is None: