Build dataset by the given config.
(cfg: Union[dict, list, tuple],
default_args: Optional[Union[dict, None]] = None)
| 26 | |
| 27 | |
| 28 | def build_dataset(cfg: Union[dict, list, tuple], |
| 29 | default_args: Optional[Union[dict, None]] = None): |
| 30 | """"Build dataset by the given config.""" |
| 31 | from .dataset_wrappers import ( |
| 32 | ConcatDataset, |
| 33 | RepeatDataset, |
| 34 | ) |
| 35 | if isinstance(cfg, (list, tuple)): |
| 36 | dataset = ConcatDataset([build_dataset(c, default_args) for c in cfg]) |
| 37 | elif cfg['type'] == 'RepeatDataset': |
| 38 | dataset = RepeatDataset( |
| 39 | build_dataset(cfg['dataset'], default_args), cfg['times']) |
| 40 | else: |
| 41 | dataset = build_from_cfg(cfg, DATASETS, default_args) |
| 42 | |
| 43 | return dataset |
| 44 | |
| 45 | |
| 46 | def build_dataloader(dataset: Dataset, |