From 13d2f989cebbebf9113cf9dfe1b8fdc294708b45 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 26 Jan 2021 17:59:54 +0100
Subject: [PATCH] Improved the function to replace array values such that it
handles also string datatypes.
---
pysegcnn/core/utils.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pysegcnn/core/utils.py b/pysegcnn/core/utils.py
index 5556b62..d0c39d6 100644
--- a/pysegcnn/core/utils.py
+++ b/pysegcnn/core/utils.py
@@ -2217,8 +2217,8 @@ def array_replace(array, lookup):
"""
# create an index array to replace the values in the lookup table
- indices = np.arange(lookup[:, 0].max() + 1)
- indices[lookup[:, 0]] = lookup[:, 1]
+ indices = np.arange(np.int(lookup[:, 0].max()) + 1).astype(object)
+ indices[lookup[:, 0].astype(np.int)] = lookup[:, 1]
# the array with the replaced values
return indices[array]
--
GitLab