From bcdb8f8b5d66d9462dfe88ccb5ca59434d9cccd1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 6 Oct 2023 15:04:46 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/dataset.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index 034e3bdf..bbe2170c 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)
--
GitLab