From 27a548c0192898a33e8b81c563f1c655d82c8d0e Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 24 May 2020 11:56:39 -0600 Subject: [PATCH] Enable blurring via settings --- codes/data/LQGT_dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py index 2f9820ef..bc5f8063 100644 --- a/codes/data/LQGT_dataset.py +++ b/codes/data/LQGT_dataset.py @@ -142,6 +142,9 @@ 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']: + img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), 10) + if self.opt['color']: # change color space if necessary img_LQ = util.channel_convert(C, self.opt['color'], [img_LQ])[0] # TODO during val no definition