From 5b506fa012b1dd0dbd6e16c6a5620f83fd2803b7 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 18 Feb 2021 16:07:28 +0100
Subject: [PATCH] Squeeze is required for multi-dimensional arrays.
---
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 502551e..695b231 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2538,9 +2538,9 @@ class NetworkInference(BaseConfig):
batch + 1, len(self.dataloader))
# check if tensor is on gpu and convert to numpy array
- inputs = inputs.cpu().numpy().squeeze(axis=0)
- labels = labels.cpu().numpy().squeeze(axis=0)
- prdctn = prdctn.cpu().numpy().squeeze(axis=0)
+ inputs = inputs.cpu().numpy().squeeze()
+ labels = labels.cpu().numpy().squeeze()
+ prdctn = prdctn.cpu().numpy().squeeze()
# check whether to reconstruct the scenes of a dataset
if self.predict_scene:
--
GitLab