From cbe09459bfe79f2394b1413876b6fc8240431b33 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Feb 2021 16:47:21 +0100
Subject: [PATCH] Use Float32 to save memory.
---
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 b3678f4..7ed0bb5 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))) 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