snapshot...

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