(cfg, )
| 379 | |
| 380 | |
| 381 | def build_eval_dataloader(cfg, ): |
| 382 | dataloaders = [] |
| 383 | cfg = copy.deepcopy(cfg) |
| 384 | for dataset_name in cfg['DATASETS']['TEST']: |
| 385 | cfg = get_config_from_name(cfg, dataset_name) |
| 386 | # adjust mapper according to dataset |
| 387 | if dataset_name == 'imagenet_val': |
| 388 | mapper = ImageNetDatasetMapper(cfg, False) |
| 389 | elif 'coco' in dataset_name: |
| 390 | mapper = CoCoInferenceDatasetMapper(cfg, False) |
| 391 | elif dataset_name == 'sam_val': |
| 392 | mapper = SamBaselineDatasetMapper(cfg, False) |
| 393 | elif dataset_name == 'sam_minival': |
| 394 | mapper = SamBaselineDatasetMapperJSON(cfg, False) |
| 395 | else: |
| 396 | mapper = None |
| 397 | if dataset_name == 'pascal_part_val_interactive': |
| 398 | mapper = PascalInstanceNewBaselineDatasetMapper(cfg, True) |
| 399 | # dataloaders += [build_custom_test_loader(cfg, dataset_name, mapper=mapper)] |
| 400 | dataloaders += [build_detection_test_loader(cfg, dataset_name, mapper=mapper)] |
| 401 | else: |
| 402 | dataloaders += [build_detection_test_loader(cfg, dataset_name, mapper=mapper)] |
| 403 | return dataloaders |
| 404 | |
| 405 | |
| 406 | def build_train_dataloader(cfg, ): |
no test coverage detected