From 1b05283ffc6cdcbcdfbe7edf025adf25fd8b3dfe Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 7 Aug 2023 14:57:55 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/esrgan_exp.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py index 0f3d3ac2..a53675d5 100644 --- a/modules/deeplearning/esrgan_exp.py +++ b/modules/deeplearning/esrgan_exp.py @@ -9,9 +9,11 @@ import h5py from util.util import scale, descale, get_grid_values_all from util.setup import home_dir +target_param = 'cld_opd_dcomp' + # SAVED_MODEL_PATH = "https://tfhub.dev/captain-pool/esrgan-tf2/1" # SAVED_MODEL_PATH = '/Users/tomrink/Downloads/esrgan-tf2_1' -SAVED_MODEL_PATH = home_dir+'esrgan-tf2_1' +SAVED_MODEL_PATH = home_dir+'/esrgan-tf2_1' model = hub.load(SAVED_MODEL_PATH) @@ -27,7 +29,7 @@ def get_image(in_file): # bt = get_grid_values_all(h5f, 'temp_11_0um_nom') # bt = bt[s_y, s_x] - cld_opd = get_grid_values_all(h5f, 'cld_opd_dcomp') + cld_opd = get_grid_values_all(h5f, target_param) cld_opd = cld_opd[s_y, s_x] shape = cld_opd.shape -- GitLab