Esempio n. 1
0
    def _evaluation_dataloader(self):
        # TODO: Support subsample n images
        assert len(self.cfg.DATASETS.TEST)

        dataloaders = []
        for dataset_name in self.cfg.DATASETS.TEST:
            dataloaders.append(
                Detectron2GoRunner.build_detection_test_loader(self.cfg, dataset_name)
            )

        self._reset_dataset_evaluators()
        return dataloaders
Esempio n. 2
0
 def build_detection_test_loader(cfg, dataset_name, mapper=None):
     return Detectron2GoRunner.build_detection_test_loader(
         cfg, dataset_name, mapper)