From 83eaba9d86cc060dc6a8613450f823111489190f Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 14 Jan 2021 17:44:37 +0100
Subject: [PATCH] Improved representation logging.
---
pysegcnn/core/dataset.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index fdd0df9..596b731 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -229,7 +229,8 @@ class ImageDataset(Dataset):
# remove labels to merge from dataset instance labels
for k, v in self.merge_labels.items():
self._labels.pop(getattr(self.label_class, k).id)
- LOGGER.info('Merging label: {} -> {}.'.format(k, v))
+ LOGGER.info('{}, merging label: {} -> {}.'.format(
+ self.__class__.__name__, k, v))
# create model class labels
# map the original class labels to an ascending sequence of integers
--
GitLab