deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enable blurring via settings

Browse Source
This commit is contained in:
James Betker 2020-05-24 11:56:39 -06:00
parent 3c2e5a0250
commit 27a548c019
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/data/LQGT_dataset.py
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