snapshot..

modules/deeplearning/srcnn_l1b_l2.py

@@ -664,7 +664,7 @@ class SRCNN:
         labels_denorm = denormalize(labels, label_param, mean_std_dct)
         preds_denorm = denormalize(preds, label_param, mean_std_dct)
 
-        # return labels_denorm, preds_denorm
+        return labels_denorm, preds_denorm
 
     def do_evaluate(self, data, ckpt_dir):
 
@@ -700,7 +700,7 @@ class SRCNN:
         self.build_model()
         self.build_training()
         self.build_evaluation()
-        self.restore(ckpt_dir)
+        return self.restore(ckpt_dir)
 
     def run_evaluate(self, data, ckpt_dir):
         self.num_data_samples = 80000
@@ -712,7 +712,7 @@ class SRCNN:
 
 def run_restore_static(directory, ckpt_dir):
     nn = SRCNN()
-    nn.run_restore(directory, ckpt_dir)
+    return nn.run_restore(directory, ckpt_dir)
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):