From f9f81c2d3f2f5d90a7f2c030a1c6e69da5e68136 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 7 Aug 2023 15:50:13 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/esrgan_exp.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py
index 007874e5..97c722e6 100644
--- a/modules/deeplearning/esrgan_exp.py
+++ b/modules/deeplearning/esrgan_exp.py
@@ -76,7 +76,7 @@ def preprocess_image(hr_image):
return tf.expand_dims(hr_image, 0)
-def run(in_file):
+def run(in_file, out_file):
# model = hub.load(SAVED_MODEL_PATH)
t0 = time.time()
hr_image = get_image(in_file)
@@ -90,4 +90,7 @@ def run(in_file):
print('inference time: ', (t1-t0))
fake_image = tf.squeeze(fake_image)
- return fake_image
\ No newline at end of file
+ if out_file is not None:
+ np.save(out_file, fake_image)
+ else:
+ return fake_image
--
GitLab