Enable blurring via settings

This commit is contained in:
James Betker 2020-05-24 11:56:39 -06:00
parent 3c2e5a0250
commit 27a548c019

View File

@ -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