From 6900e98eb23539e9169e6728d18ddc6dad98f0b7 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 15:50:59 +0100
Subject: [PATCH] Improved logging for inference.
---
pysegcnn/core/trainer.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 271f77f..4fc4f2f 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2674,9 +2674,12 @@ class NetworkInference(BaseConfig):
# check whether to aggregate the results of the different model runs
if self.aggregate:
+ LOGGER.info('Aggregating statistics of models:')
+ LOGGER.info(('\n ' + (len(__name__) + 1) * ' ').join(
+ ['{}'.format(mstate.name) for mstate in self.state_files]))
# base name for all models
- base_name = str(self.state_files[0]).name
+ base_name = str(self.state_files[0].name)
# chech whether to compute the aggregated confusion matrix
if self.cm:
--
GitLab