From 8f6b2d2ac3fb31f7ae8ed4c6c5415269ae9f94de Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Jan 2021 14:02:46 +0100
Subject: [PATCH] Handle 2d-input arrays.
---
pysegcnn/core/utils.py | 21 +++++++++++++++++----
1 file changed, 17 insertions(+), 4 deletions(-)
diff --git a/pysegcnn/core/utils.py b/pysegcnn/core/utils.py
index e249e07..4f1b716 100644
--- a/pysegcnn/core/utils.py
+++ b/pysegcnn/core/utils.py
@@ -137,10 +137,23 @@ def img2np(path, tile_size=None, tile=None, pad=False, cval=0):
# accept numpy arrays as input
elif isinstance(path, np.ndarray):
+ # input array
img = path
- bands = img.shape[0]
- height = img.shape[1]
- width = img.shape[2]
+
+ # check the dimensions of the input array
+ if img.ndim > 2:
+ bands = img.shape[0]
+ height = img.shape[1]
+ width = img.shape[2]
+ else:
+ bands = 1
+ height = img.shape[0]
+ width = img.shape[1]
+
+ # expand input array to fit band dimension
+ img = np.expand_dims(img, axis=0)
+
+ # input array data type
dtype = img.dtype
else:
@@ -728,7 +741,7 @@ def reconstruct_scene(tiles):
# convert to numpy array
tiles = np.asarray(tiles)
- # check the size
+ # check the dimensions of the input array
if tiles.ndim > 3:
nbands = tiles.shape[1]
tile_size = tiles.shape[2]
--
GitLab