diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index 034e3bdf60b72637e3528353e995a0148d7d36ec..bbe2170c79d9257105794440683ec142be7148f0 100644
--- a/modules/GSOC/E2_ESRGAN/lib/dataset.py
+++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py
@@ -197,11 +197,12 @@ class OpdNpyDataset:
         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.math.multiply(low_resolution, 255.0)
-        low_resolution = tf.clip_by_value(low_resolution, 0, 255)
+        # low_resolution = tf.math.multiply(low_resolution, 255.0)
+        low_resolution = tf.clip_by_value(low_resolution, 0, 1)
 
-        hr_image = tf.math.multiply(hr_image[:, :, :, 1], 255.0)
-        high_resolution = tf.clip_by_value(hr_image, 0, 255)
+        # hr_image = tf.math.multiply(hr_image[:, :, :, 1], 255.0)
+        hr_image = hr_image[:, :, :, 1]
+        high_resolution = tf.clip_by_value(hr_image, 0, 1)
         high_resolution = tf.expand_dims(high_resolution, axis=3)
 
         low_resolution, high_resolution = augment_image()(low_resolution, high_resolution)