From 31cf1ac98dee3d813ce6244eb7c3074ae7b833fa Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 27 Oct 2020 10:26:19 -0600 Subject: [PATCH] Retrofit full_image_dataset to work with new arch. --- codes/data/full_image_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/data/full_image_dataset.py b/codes/data/full_image_dataset.py index 84893699..6fe2d444 100644 --- a/codes/data/full_image_dataset.py +++ b/codes/data/full_image_dataset.py @@ -303,9 +303,9 @@ class FullImageDataset(data.Dataset): gt_fullsize_ref = cv2.cvtColor(gt_fullsize_ref, cv2.COLOR_BGR2RGB) # LQ needs to go to a PIL image to perform the compression-artifact transformation. - if self.opt['phase'] == 'train': - img_LQ = self.pil_augment(img_LQ) - lq_fullsize_ref = self.pil_augment(lq_fullsize_ref, strength=.2) + #if self.opt['phase'] == 'train': + #img_LQ = self.pil_augment(img_LQ) + #lq_fullsize_ref = self.pil_augment(lq_fullsize_ref, strength=.2) img_GT = torch.from_numpy(np.ascontiguousarray(np.transpose(img_GT, (2, 0, 1)))).float() gt_fullsize_ref = torch.from_numpy(np.ascontiguousarray(np.transpose(gt_fullsize_ref, (2, 0, 1)))).float()