Commit 5600f01e authored by Kai Zhang's avatar Kai Zhang Committed by Facebook GitHub Bot
Browse files

allow to override get_evaluator

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/50

Found a few custom runners override this method of default runner.

Reviewed By: zhanghang1989, wat3rBro

Differential Revision: D27777505

fbshipit-source-id: 0463cc36bad4af4cbfbe09ab46962cfc1dafbd5d
parent c6c089a6
...@@ -282,13 +282,19 @@ class DefaultTask(pl.LightningModule): ...@@ -282,13 +282,19 @@ class DefaultTask(pl.LightningModule):
output_folder = _get_inference_dir_name( output_folder = _get_inference_dir_name(
self.cfg.OUTPUT_DIR, "inference", dataset_name, tag self.cfg.OUTPUT_DIR, "inference", dataset_name, tag
) )
evaluator = Detectron2GoRunner.get_evaluator( evaluator = self.get_evaluator(
self.cfg, dataset_name, output_folder=output_folder self.cfg, dataset_name, output_folder=output_folder
) )
evaluator.reset() evaluator.reset()
dataset_evaluators.append(evaluator) dataset_evaluators.append(evaluator)
# TODO: add visualization evaluator # TODO: add visualization evaluator
@staticmethod
def get_evaluator(cfg: CfgNode, dataset_name: str, output_folder: str):
return Detectron2GoRunner.get_evaluator(
cfg=cfg, dataset_name=dataset_name, output_folder=output_folder
)
@staticmethod @staticmethod
def build_detection_test_loader(cfg, dataset_name, mapper=None): def build_detection_test_loader(cfg, dataset_name, mapper=None):
return Detectron2GoRunner.build_detection_test_loader(cfg, dataset_name, mapper) return Detectron2GoRunner.build_detection_test_loader(cfg, dataset_name, mapper)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment