From 7fdeea93ea4dc265672d517e5684eedb225e77b2 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 6 Oct 2023 13:33:59 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/esrgan_exp.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index 7ea93974..d43dc26a 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -6,7 +6,7 @@ import tensorflow as tf
#import tensorflow_hub as hub
import matplotlib.pyplot as plt
import h5py
-from util.util import scale, descale, get_grid_values_all
+from util.util import descale2, get_grid_values_all
from util.setup import home_dir
target_param = 'cld_opd_dcomp'
@@ -123,6 +123,13 @@ 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
+ hr_image = descale2(hr_image[:, :, :, 1], 0.0, 160.0)
+ image = descale2(image[:, :, :, 1], 0.0, 160.0)
+
+ sres_image = descale2(sres_image, 0.0, 160.0)
+
if out_file is not None:
# np.save(out_file, sres_image)
with open(out_file, 'wb') as f:
--
GitLab