Function _evaluate

tests/core/inferenceutils/test_map_computation.py:251–284

(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])

    gt_assemblies = _to_assemblies(gt, ground_truth=True)
    pred_assemblies = _to_assemblies(pred, ground_truth=False)
    oks = inferenceutils.evaluate_assembly_greedy(
        assemblies_gt=gt_assemblies,
        assemblies_pred=pred_assemblies,
        oks_sigma=0.1,
        oks_thresholds=np.linspace(0.5, 0.95, 10),
        margin=0.0,
        symmetric_kpts=None,
    )

    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()
    print(20 * "-")
    print(f"pycocotools mAP: {coco_oks}")
    print()
    assert oks["mAP"] == coco_oks


def _to_assemblies(