diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index e5db230d11bcefc69858efd35a8ce089ee67489b..e8d2f87c5f5097776bd6ec0b0c69615ecec3ffef 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -661,7 +661,7 @@ def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'): return nda_lr -def run_evaluate_static(in_file, out_file, param='temp_11_0um_nom', ckpt_dir='/Users/tomrink/tf_model_sres/run-20220805173619/'): +def run_evaluate_static(in_file, out_file, ckpt_dir): nda = np.load(in_file) nda = nda[:, data_idx, 3:131:2, 3:131:2] @@ -669,7 +669,7 @@ def run_evaluate_static(in_file, out_file, param='temp_11_0um_nom', ckpt_dir='/U nda = np.expand_dims(nda, axis=3) nn = SRCNN() - out_sr = nn.run_evaluate(nda, param, ckpt_dir) + out_sr = nn.run_evaluate(nda, data_param, ckpt_dir) if out_file is not None: np.save(out_file, out_sr) else: