Function _evaluate

tests/core/metrics/test_metrics_map_computation.py:253–278

(gt: dict[str, np.ndarray], pred: dict[str, np.ndarray])

def _evaluate(gt: dict[str, np.ndarray], pred: dict[str, np.ndarray]):
    for k, v in gt.items():
        print(20 * "-")
        print(k)
        print("GT")
        print(v)
        print("PR")
        print(pred[k])

    data = prepare_evaluation_data(gt, pred)
    oks = compute_oks(data, oks_bbox_margin=0)

    num_joints = gt[list(gt.keys())[0]].shape[1]
    coco_gt = _to_coco_ground_truth(gt, num_joints, bbox_margin=0)
    coco_pred = _to_coco_predictions(coco_gt, pred, bbox_margin=0)
    coco_oks = eval_coco(coco_gt, coco_pred, num_joints)
    print(20 * "-")
    print("dlc mAP:")
    for k, v in oks.items():
        print(k)
        print(v)
    print(20 * "-")
    print(f"pycocotools mAP: {coco_oks}")
    print()
    dlc_map = oks["mAP"] / 100
    assert_almost_equal(dlc_map, coco_oks)


def _to_coco_ground_truth(