def main()

in scripts/eval_grd_anet_entities.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

def main(args):

    grd_evaluator = ANetGrdEval(reference_file=args.reference, submission_file=args.submission,
                           split_file=args.split_file, val_split=args.split,
                           iou_thresh=args.iou_thresh, verbose=args.verbose)
    if args.eval_mode == 'GT':
        print('Assuming the input boxes are based upon GT sentences.')
        grd_evaluator.gt_grd_eval()
    elif args.eval_mode == 'gen':
        print('Assuming the input boxes are based upon generated sentences.')
        grd_evaluator.grd_eval(mode=args.loc_mode)
    else:
        raise Exception('Invalid eval mode!')