hotr/engine/__init__.py

from .evaluator_vcoco import vcoco_evaluate, vcoco_accumulate
from .evaluator_hico import hico_evaluate

def hoi_evaluator(args, model, criterion, postprocessors, data_loader, device, thr=0):
    if args.dataset_file == 'vcoco':
        return vcoco_evaluate(model, criterion, postprocessors, data_loader, device, args.output_dir, thr,args=args)
    elif args.dataset_file == 'hico-det':
        return hico_evaluate(model, postprocessors, data_loader, device, thr,args=args)
    else: raise NotImplementedError

def hoi_accumulator(args, total_res, print_results=False, wandb=False):
    if args.dataset_file == 'vcoco':
        return vcoco_accumulate(total_res, args, print_results, wandb)
    else: raise NotImplementedError