a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests 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 img_PIX = img_GT
# get LQ image # get LQ image
already_blurred = False
if self.paths_LQ: if self.paths_LQ:
LQ_path = self.get_lq_path(index) LQ_path = self.get_lq_path(index)
resolution = [int(s) for s in self.sizes_LQ[index].split('_') 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) img_GT = cv2.cvtColor(img_GT, cv2.COLOR_GRAY2BGR)
H, W, _ = img_GT.shape 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 # 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: if img_LQ.ndim == 2:
img_LQ = np.expand_dims(img_LQ, axis=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'], img_LQ, img_GT, img_PIX = util.augment([img_LQ, img_GT, img_PIX], self.opt['use_flip'],
self.opt['use_rot']) self.opt['use_rot'])
if self.opt['use_blurring']: if self.opt['use_blurring'] and not already_blurred:
blur_sig = int(random.randrange(0, 10)) blur_sig = int(random.randrange(0, 4))
img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), blur_sig) img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), blur_sig)
if self.opt['color']: # change color space if necessary if self.opt['color']: # change color space if necessary