From 72a3bc7c63b1d3230ec4f787001b5282ff49a57d Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Feb 2021 17:37:39 +0100
Subject: [PATCH] Use Float16 to save memory.
---
pysegcnn/core/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index 7ed0bb5..ba9df96 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -634,7 +634,7 @@ 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.float32)
+ 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'}
# iterate over the samples of the dataset
--
GitLab