snapshot...

modules/deeplearning/srcnn_l1b_l2.py

@@ -652,6 +652,11 @@ class SRCNN:
         return self.do_evaluate(data, ckpt_dir)
 
 
+def run_restore_static(directory, ckpt_dir):
+    nn = SRCNN()
+    nn.run_restore(directory, ckpt_dir)
+
+
 def run_evaluate_static(in_file, out_file, ckpt_dir):
     h5f = h5py.File(in_file, 'r')
     grd_a = get_grid_values_all(h5f, data_params[0])
@@ -700,7 +705,7 @@ def analyze():
 
     h5f.close()
 
-    return grd, grd_hr
+    return grd, grd_lr, grd_hr
 
 
 if __name__ == "__main__":