From 135cafe52e83638f9a635c1a2f98958424e89372 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 18 Feb 2021 16:21:37 +0100
Subject: [PATCH] First check if an entire scene is already processed before
appending.
---
pysegcnn/core/trainer.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 695b231..4e82d54 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2545,10 +2545,6 @@ class NetworkInference(BaseConfig):
# check whether to reconstruct the scenes of a dataset
if self.predict_scene:
- # append model predictions of current batch to scene dictionary
- for k, v in zip(INFERENCE_NAMES, [inputs, labels, prdctn]):
- scenes[k].append(v)
-
# check if an entire scene is processed
if batch % self.trg_ds.dataset.tiles == 0 and batch != 0:
@@ -2607,6 +2603,10 @@ class NetworkInference(BaseConfig):
self.scenes_path.joinpath(batch_name)),
bbox_inches='tight')
+ # append model predictions of current batch to scene dictionary
+ for k, v in zip(INFERENCE_NAMES, [inputs, labels, prdctn]):
+ scenes[k].append(v)
+
else:
# save current batch to output dictionary
output[batch] = {k: v for k, v in zip(INFERENCE_NAMES,
--
GitLab