diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py index fdd0df9cb50d2f422ce54e52b4e8a23e560abd26..596b731c8ab0fba9ae2c39cdb58ba7f0dbe5d5c9 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