From 7cce87b33b69d22583b4382b9ace915e6f9e6e27 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 11 Aug 2020 16:40:52 +0200
Subject: [PATCH] Removed redundant dataset names
---
pysegcnn/core/dataset.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index a3371c0..ee36ba3 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -668,10 +668,10 @@ class Cloud95Dataset(ImageDataset):
class SupportedDatasets(enum.Enum):
- Sparcs = {'name': 'Sparcs', 'class': SparcsDataset}
- Cloud95 = {'name': 'Cloud95', 'class': Cloud95Dataset}
- Garmisch = {'name': 'Garmisch', 'class': ProSnowGarmisch}
- Obergurgl = {'name': 'Obergurgl', 'class': ProSnowObergurgl}
+ Sparcs = SparcsDataset
+ Cloud95 = Cloud95Dataset
+ Garmisch = ProSnowGarmisch
+ Obergurgl = ProSnowObergurgl
if __name__ == '__main__':
--
GitLab