snapshot...

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: