From 5e28a1c8d75b95e3da69504de02a0d37ce913ac8 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Mon, 20 Jul 2020 17:10:30 +0200
Subject: [PATCH] Added an enumeration of the currently supported datasets
---
pytorch/constants.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/pytorch/constants.py b/pytorch/constants.py
index 6f7cd56..c536806 100644
--- a/pytorch/constants.py
+++ b/pytorch/constants.py
@@ -5,8 +5,12 @@ Created on Tue Jul 14 10:58:20 2020
@author: Daniel
"""
# builtins
+from __future__ import absolute_import
import enum
+# locals
+from pytorch.dataset import (SparcsDataset, Cloud95Dataset, ProSnowGarmisch,
+ ProSnowObergurgl)
# Landsat 8 bands
class Landsat8(enum.Enum):
@@ -62,3 +66,10 @@ class ProSnowLabels(enum.Enum):
Land = 0, 'grey'
Cloud = 1, 'white'
Snow = 2, 'lightblue'
+
+
+class SupportedDatasets(enum.Enum):
+ Sparcs = {'name': 'Sparcs', 'class': SparcsDataset}
+ Cloud95 = {'name': 'Cloud95', 'class': Cloud95Dataset}
+ Garmisch = {'name': 'Garmisch', 'class': ProSnowGarmisch}
+ Obergurgl = {'name': 'Obergurgl', 'class': ProSnowObergurgl}
--
GitLab