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

Randomize blur effect

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

3
codes/data/LQGT_dataset.py
View File

@ -143,7 +143,8 @@ class LQGTDataset(data.Dataset):
self.opt['use_rot']) self.opt['use_rot'])
if self.opt['use_blurring']: if self.opt['use_blurring']:
img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), 10) blur_sig = int(random.randrange(0, 10))
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
img_LQ = util.channel_convert(C, self.opt['color'], img_LQ = util.channel_convert(C, self.opt['color'],