diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 16be62a3f5b164fed59f44ef57335bf4c30c3846..2b075def8ed2799624e53b58b8089377283765fe 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -657,6 +657,15 @@ class SRCNN: print('loss, acc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) + labels = np.concatenate(self.test_labels) + preds = np.concatenate(self.test_preds) + print(labels.shape, preds.shape) + + labels_denorm = denormalize(labels, label_param, mean_std_dct) + preds_denorm = denormalize(preds, label_param, mean_std_dct) + + # return labels_denorm, preds_denorm + def do_evaluate(self, data, ckpt_dir): ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)