diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py index e3fa5eaf07273d2cd7930daab8ec25dc6879fd7f..438fa21607dc1a2337f94e399cb83dfaa3b4b43e 100644 --- a/modules/GSOC/E2_ESRGAN/lib/dataset.py +++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py @@ -196,7 +196,10 @@ class OpdNpyDataset: hr_image = np.stack([refl, opd], axis=3) hr_image = tf.image.crop_to_bounding_box(hr_image, 0, 0, self.hr_size, self.hr_size) - low_resolution = tf.image.resize(hr_image, [self.lr_size, self.lr_size], method='bicubic') + # low_resolution = tf.image.resize(hr_image, [self.lr_size, self.lr_size], method='bicubic') + # low_resolution = tf.image.resize(hr_image, [self.lr_size, self.lr_size], method='nearest') + low_resolution = hr_image[:, ::4, ::4, :] + low_resolution = low_resolution.copy() low_resolution = tf.math.multiply(low_resolution, 255.0) low_resolution = tf.clip_by_value(low_resolution, 0, 255.0)