From 984a1c541c85a9eaebde9c831624b73c363e078f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 8 Oct 2023 11:58:13 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/dataset.py | 9 ++++-----
1 file changed, 4 insertions(+), 5 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index bbe2170c..e3fa5eaf 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)
--
GitLab