fix test dataset loading with proposal enabled
Summary: (https://github.com/facebookresearch/detectron2/commit/60fd4885d7cfd52d4267d1da9ebb6b2b9a3fc937) changed the dataset name type while load proposals were still expecting single item inputs.
@@ -404,7 +404,8 @@ def _test_loader_from_config(cfg, dataset_name, mapper=None):
dataset_name,
filter_empty=False,
proposal_files=[
- cfg.DATASETS.PROPOSAL_FILES_TEST[list(cfg.DATASETS.TEST).index(dataset_name)]
+ cfg.DATASETS.PROPOSAL_FILES_TEST[list(cfg.DATASETS.TEST).index(x)]
+ for x in dataset_name
]
if cfg.MODEL.LOAD_PROPOSALS
else None,
