diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py index 0f3d3ac2146511f1bad4ce0000c0e6f9af077151..a53675d5e038f3fbf0df700e2f047f5e48940723 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