From 42efced58ee864b01f588f139ab77c1acb3ca127 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Mon, 17 Aug 2020 17:20:13 +0200
Subject: [PATCH] Cval is now correctly initialized
---
pysegcnn/core/dataset.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index 011d94c..98a2428 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -132,6 +132,7 @@ class ImageDataset(Dataset):
self.transforms = [None] + self.transforms
# when padding, add a new "no data" label to the ground truth
+ self.cval = 0
if self.pad and sum(self.padding) > 0:
self.cval = max(self.labels) + 1
self.labels[self.cval] = {'label': 'No data', 'color': 'black'}
--
GitLab