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