From e111174b29c45c5ec806aff157754d6894026525 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Sun, 19 Jan 2020 12:01:17 -0500
Subject: [PATCH] [common] (roco dset) add enum for roco dsets

---
 .../research_common/dataset/roco/roco_datasets.py  | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 common/research_common/dataset/roco/roco_datasets.py

diff --git a/common/research_common/dataset/roco/roco_datasets.py b/common/research_common/dataset/roco/roco_datasets.py
new file mode 100644
index 0000000..dde91d3
--- /dev/null
+++ b/common/research_common/dataset/roco/roco_datasets.py
@@ -0,0 +1,14 @@
+from enum import Enum
+
+from research_common.constants import ROCO_DSET_DIR
+from research_common.dataset.dataset import Dataset
+
+
+class ROCODataset(Dataset, Enum):
+    def __init__(self, competition_name: str):
+        super().__init__(ROCO_DSET_DIR / competition_name, self.name)
+
+    CentralChina = "robomaster_Central China Regional Competition"
+    NorthChina = "robomaster_North China Regional Competition"
+    SouthChina = "robomaster_South China Regional Competition"
+    Final = "robomaster_Final Tournament"
-- 
GitLab