From b075a5e7142de9e2eeda66769e4c88a0919f9897 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Feb 2021 18:33:45 +0100
Subject: [PATCH] Fixed bug: No data class is now correctly skipped.
---
pysegcnn/core/dataset.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index ba9df96..b4ed855 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -634,8 +634,8 @@ class ImageDataset(Dataset):
# initialize dictionary of class spectral distribution
# exclude NoData class
- cls_ds = {k: np.empty(shape=(0, len(self.use_bands)), dtype=np.float16)
- for k, v in self.labels.items() if v['label'] != 'No_data'}
+ cls_ds = {k: np.empty(shape=(0, len(self.use_bands)), dtype=np.float32)
+ for k, v in self.labels.items() if v['label'] != 'No data'}
# iterate over the samples of the dataset
for i in range(len(self)):
--
GitLab