test_metrics.py 文件源码

def test_box_avg_precision():
    boxes1 = [(1, 1, 3, 3), (4, 2, 2, 3), (5, 5, 2, 1)]
    scores1 = [0.5, 0.2, 0.1]
    boxes2 = [(2, 1, 2, 3), (4, 3, 2, 3)]
    scores2 = [0.5, 0.2]

    ap = nm.box_avg_precision(boxes2, boxes2, scores2)
    assert ap == 1.0

    ap = nm.box_avg_precision(boxes1, boxes2, scores2)
    assert ap == approx(0.63, abs=1e-2)

    ap = nm.box_avg_precision(boxes2, boxes1, scores1)
    assert ap == 1.0

    ap = nm.box_avg_precision(boxes1, [], [])
    assert ap == 0.0

    ap = nm.box_avg_precision([], boxes1, scores1)
    assert ap == 0.0