File size: 1,377 Bytes
3bbb319 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import numpy as np
from mmdet.core.evaluation.recall import eval_recalls
det_bboxes = np.array([
[0, 0, 10, 10],
[10, 10, 20, 20],
[32, 32, 38, 42],
])
gt_bboxes = np.array([[0, 0, 10, 20], [0, 10, 10, 19], [10, 10, 20, 20]])
gt_ignore = np.array([[5, 5, 10, 20], [6, 10, 10, 19]])
def test_eval_recalls():
gts = [gt_bboxes, gt_bboxes, gt_bboxes]
proposals = [det_bboxes, det_bboxes, det_bboxes]
recall = eval_recalls(
gts, proposals, proposal_nums=2, use_legacy_coordinate=True)
assert recall.shape == (1, 1)
assert 0.66 < recall[0][0] < 0.667
recall = eval_recalls(
gts, proposals, proposal_nums=2, use_legacy_coordinate=False)
assert recall.shape == (1, 1)
assert 0.66 < recall[0][0] < 0.667
recall = eval_recalls(
gts, proposals, proposal_nums=2, use_legacy_coordinate=True)
assert recall.shape == (1, 1)
assert 0.66 < recall[0][0] < 0.667
recall = eval_recalls(
gts,
proposals,
iou_thrs=[0.1, 0.9],
proposal_nums=2,
use_legacy_coordinate=False)
assert recall.shape == (1, 2)
assert recall[0][1] <= recall[0][0]
recall = eval_recalls(
gts,
proposals,
iou_thrs=[0.1, 0.9],
proposal_nums=2,
use_legacy_coordinate=True)
assert recall.shape == (1, 2)
assert recall[0][1] <= recall[0][0]
|