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 0000000000000000000000000000000000000000..dde91d324b16f110599650bdd5d1f143da3b55c6 --- /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"