Python Detectron2GoRunner.build_detection_test_loader Examples

Example #1

    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

Example #2

File: lightning_task.py Project: guowenbin90/d2go

 def build_detection_test_loader(cfg, dataset_name, mapper=None):
     return Detectron2GoRunner.build_detection_test_loader(
         cfg, dataset_name, mapper)