From 69cbfa2f0c57781437514911774eb75bed97cde0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 26 May 2020 13:48:34 -0600 Subject: [PATCH] Adjust dataset mutations a bit Adjusts the compression artfacts to be more aggressive, and blurring to be less aggressive. --- codes/data/LQGT_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py index 02583f92..51949ef0 100644 --- a/codes/data/LQGT_dataset.py +++ b/codes/data/LQGT_dataset.py @@ -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)