From bf8e608bcd9d3f58bac2db9ecd4ee3893fb500f8 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Mar 2021 14:43:11 +0100
Subject: [PATCH] Copy Tensor to cpu before mapping to target labels.
---
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 206e780..a32fa26 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2532,10 +2532,6 @@ class NetworkInference(BaseConfig):
prdctn = F.softmax(
model(inputs), dim=1).argmax(dim=1).squeeze()
- # check whether the source and target domain labels differ
- if self.apply_label_map:
- prdctn = self.map_to_target(prdctn)
-
# progress string to log
progress = 'Sample: {:d}/{:d}'.format(
batch + 1, len(self.dataloader))
@@ -2545,6 +2541,10 @@ class NetworkInference(BaseConfig):
labels = labels.cpu().numpy().squeeze()
prdctn = prdctn.cpu().numpy().squeeze()
+ # check whether the source and target domain labels differ
+ if self.apply_label_map:
+ prdctn = self.map_to_target(prdctn)
+
# check whether to reconstruct the scenes of a dataset
if self.predict_scene:
--
GitLab