From e0cbf63e90e8fded7690aded1524ae1fd32fa270 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 8 Oct 2023 11:03:24 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/esrgan_exp.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index d43dc26a..d4e4436d 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -101,7 +101,7 @@ def preprocess_image(hr_image):
hr_image = tmp_image
hr_image *= 255.0
- hr_image *= 255.0
+ # hr_image *= 255.0
hr_size = (tf.convert_to_tensor(hr_image.shape[:-1]) // 4) * 4
hr_image = tf.image.crop_to_bounding_box(hr_image, 0, 0, hr_size[0], hr_size[1])
@@ -123,11 +123,12 @@ def run(in_file, out_file):
print('inference time: ', (t1-t0))
sres_image = tf.squeeze(sres_image)
- sres_image /= 255.0
- hr_image /= 255.0
+ # sres_image /= 255.0
+ # hr_image /= 255.0
hr_image = descale2(hr_image[:, :, :, 1], 0.0, 160.0)
image = descale2(image[:, :, :, 1], 0.0, 160.0)
+ sres_image = sres_image.numpy()
sres_image = descale2(sres_image, 0.0, 160.0)
if out_file is not None:
--
GitLab