Build dataset from options. Args: dataset_opt (dict): Configuration for dataset. It must contain: name (str): Dataset name. type (str): Dataset type.
(dataset_opt)
| 23 | |
| 24 | |
| 25 | def build_dataset(dataset_opt): |
| 26 | """Build dataset from options. |
| 27 | |
| 28 | Args: |
| 29 | dataset_opt (dict): Configuration for dataset. It must contain: |
| 30 | name (str): Dataset name. |
| 31 | type (str): Dataset type. |
| 32 | """ |
| 33 | dataset_opt = deepcopy(dataset_opt) |
| 34 | dataset = DATASET_REGISTRY.get(dataset_opt['type'])(dataset_opt) |
| 35 | logger = get_root_logger() |
| 36 | logger.info(f'Dataset [{dataset.__class__.__name__}] - {dataset_opt["name"]} is built.') |
| 37 | return dataset |
| 38 | |
| 39 | |
| 40 | def build_dataloader(dataset, dataset_opt, num_gpu=1, dist=False, sampler=None, seed=None): |