From 1d54efffb32ecb60b526982f3d13e6aaeb305091 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:38:40 +0100
Subject: [PATCH] Fixed a bug: scenes are now properly indexed.
---
pysegcnn/core/trainer.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index c620073..271633e 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -955,9 +955,9 @@ class LogConfig(BaseConfig):
The string to write to the model log file.
"""
- LOGGER.info(200 * '-')
+ LOGGER.info(80 * '-')
LOGGER.info('{}: '.format(LogConfig.now()) + init_str)
- LOGGER.info(200 * '-')
+ LOGGER.info(80 * '-')
@dataclasses.dataclass
@@ -2506,7 +2506,7 @@ class NetworkInference(BaseConfig):
# id of the current scene
current_scene = np.int(batch * self.dataloader.batch_size)
- batch = self.trg_ds.dataset.scenes['id'][current_scene]
+ batch = self.trg_ds.dataset.scenes[current_scene]['id']
# modify the progress string
progress = progress.replace('Sample', 'Scene')
--
GitLab