def test_match_coco_at_150(self):
matcher = MatchEngineEuclideanDistance(150, 'coco')
assert np.all(
matcher.match(detections, gt) == np.array(
[[0, 1, 0, 0, 0, 0], [0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 1, 0],
[0, 0, 0, 1, 0, 0], [1, 0, 0, 0, 0, 0], [0, 0, 1, 0, 0, 0],
[0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0]]))
def test_match_xview_at_200(self):
matcher = MatchEngineEuclideanDistance(200, 'xview')
assert np.all(
matcher.match(detections, gt) == np.array(
[[0, 1, 0, 0, 0, 0], [0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 1, 0],
[0, 0, 0, 1, 0, 0], [1, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0], [0, 0, 1, 0, 0, 0]]))