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

Adjust dataset mutations a bit

Browse Source 
Adjusts the compression artfacts to be more aggressive, and blurring to be less aggressive.
This commit is contained in:
James Betker 2020-05-26 13:48:34 -06:00
parent 2931142458
commit 69cbfa2f0c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/data/LQGT_dataset.py
View File

@ -144,7 +144,7 @@ class LQGTDataset(data.Dataset):
self.opt['use_rot'])
if self.opt['use_blurring']:
blur_sig = int(random.randrange(0, 4))
blur_sig = int(random.randrange(0, 3))
img_LQ = cv2.GaussianBlur(img_LQ, (3, 3), blur_sig)
if self.opt['color']: # change color space if necessary
@ -161,7 +161,7 @@ class LQGTDataset(data.Dataset):
img_LQ = (img_LQ * 255).astype(np.uint8)
img_LQ = Image.fromarray(img_LQ)
if self.opt['use_compression_artifacts']:
qf = random.randrange(15, 100)
qf = random.randrange(10, 70)
corruption_buffer = BytesIO()
img_LQ.save(corruption_buffer, "JPEG", quality=qf, optimice=True)
corruption_buffer.seek(0)