__init__.py 246 B

12345678910
  1. from evaluator.coco_evaluator import COCOAPIEvaluator
  2. def build_evluator(args, cfg, device):
  3. evaluator = None
  4. # COCO Evaluator
  5. if args.dataset == 'coco':
  6. evaluator = COCOAPIEvaluator(args, cfg, device)
  7. return evaluator