From fabe980434c516725ccab5ad0d1fd46bd4338413 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 15:41:22 +0100
Subject: [PATCH] Improved logging for inference.
---
pysegcnn/core/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 8943cec..271f77f 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2652,6 +2652,7 @@ class NetworkInference(BaseConfig):
in output.items()]).flatten()
# calculate confusion matrix
+ LOGGER.info('Computing confusion matrix ...')
conf_mat = confusion_matrix(y_true, y_pred)
# add confusion matrix to model output
--
GitLab