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

Blur HR image before downsizing, when available

Browse Source
This commit is contained in:
James Betker 2020-05-24 17:18:44 -06:00
parent cc4571eb8d
commit 2f8b0250b9
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
codes/data/LQGT_dataset.py
View File

@ -90,6 +90,7 @@ class LQGTDataset(data.Dataset):
img_PIX = img_GT
# get LQ image
already_blurred = False
if self.paths_LQ:
LQ_path = self.get_lq_path(index)
resolution = [int(s) for s in self.sizes_LQ[index].split('_')
@ -113,8 +114,16 @@ class LQGTDataset(data.Dataset):
img_GT = cv2.cvtColor(img_GT, cv2.COLOR_GRAY2BGR)
H, W, _ = img_GT.shape
if self.opt['use_blurring']:
blur_sig = int(random.randrange(0, 4))
hqxform = cv2.GaussianBlur(img_GT, (3, 3), blur_sig)
already_blurred = True
else:
hqxform = img_GT
# using matlab imresize
img_LQ = util.imresize_np(img_GT, 1 / scale, True)
img_LQ = util.imresize_np(hqxform, 1 / scale, True)
if img_LQ.ndim == 2:
img_LQ = np.expand_dims(img_LQ, axis=2)
@ -142,8 +151,8 @@ class LQGTDataset(data.Dataset):
img_LQ, img_GT, img_PIX = util.augment([img_LQ, img_GT, img_PIX], self.opt['use_flip'],
self.opt['use_rot'])
if self.opt['use_blurring']:
blur_sig = int(random.randrange(0, 10))
if self.opt['use_blurring'] and not already_blurred:
blur_sig = int(random.randrange(0, 4))
img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), blur_sig)
if self.opt['color']: # change color space if necessary