| 30 | |
| 31 | |
| 32 | def build_dataset(cfg, default_args=None): |
| 33 | if isinstance(cfg, (list, tuple)): |
| 34 | dataset = ConcatDataset([build_dataset(c, default_args) for c in cfg]) |
| 35 | elif cfg['type'] == 'RepeatDataset': |
| 36 | dataset = RepeatDataset( |
| 37 | build_dataset(cfg['dataset'], default_args), cfg['times']) |
| 38 | elif isinstance(cfg['data_source'].get('ann_file'), (list, tuple)): |
| 39 | dataset = _concat_dataset(cfg, default_args) |
| 40 | else: |
| 41 | dataset = build_from_cfg(cfg, DATASETS, default_args) |
| 42 | |
| 43 | return dataset |
| 44 | |
| 45 | |
| 46 | def build_dali_dataset(cfg, default_args=None): |