def test_evaluator_iou(pred, gt):
evalDetection = EvaluatorDetection(pred, gt)
iou = evalDetection.iou()
assert iou['dog'] == 0.662
assert iou['cat'] == 0.667
def test_evaluator_AP(pred, gt):
evalDetection = EvaluatorDetection(pred, gt)
AP = evalDetection.AP()
assert round(AP['dog'], 3) == 1
assert AP['cat'] == 0.5
def test_evaluator_mean_iou(pred, gt):
evalDetection = EvaluatorDetection(pred, gt)
mean_iou = evalDetection.mean_iou()
assert mean_iou == 0.665
def test_evaluator_mAP(pred, gt):
evalDetection = EvaluatorDetection(pred, gt)
mAP = evalDetection.mAP()
assert mAP == 0.75