diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index a03d64df77cd450729b493ed22ee4d2e73f97cc5..1c706e2893f556941d12006d3cb7855ff744d3a3 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -614,19 +614,6 @@ class ESPCN: print('loss, acc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) - # labels = np.concatenate(self.test_labels) - # self.test_labels = labels - # - # preds = np.concatenate(self.test_preds) - # self.test_probs = preds - # - # if NumClasses == 2: - # preds = np.where(preds > 0.5, 1, 0) - # else: - # preds = np.argmax(preds, axis=1) - # - # self.test_preds = preds - def do_evaluate(self, nda_lr, param, ckpt_dir): ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)