| from ..base import DataBase | |
| from .dataset import BEVKitti360Dataset | |
| from ..schema import KITTIDataConfiguration | |
| class BEVKitti360Data(DataBase): | |
| def __init__(self, cfg: KITTIDataConfiguration) -> None: | |
| self.cfg = cfg | |
| self._dataset = {} | |
| def prepare_data(self) -> None: | |
| return | |
| def setup(self, stage: str) -> None: | |
| split = { | |
| 'fit': 'train', | |
| 'val': 'val', | |
| 'validate': 'val', | |
| 'test': 'val', | |
| "train": "train" | |
| }[stage] | |
| self._dataset[stage] = BEVKitti360Dataset( | |
| cfg=self.cfg, | |
| split_name=split | |
| ) | |
| def dataset(self, stage: str): | |
| if self._dataset.get(stage) is None: | |
| self.setup(stage) | |
| return self._dataset[stage] | |