def test_duplicate_cams(self):
distance_matrix = np.tile(np.arange(5), (5, 1))
result = eval_cmc(distance_matrix,
query_ids=np.array([0, 0, 0, 1, 1]),
gallery_ids=np.array([0, 0, 0, 1, 1]),
query_cams=np.array([0, 0, 0, 0, 0]),
gallery_cams=np.array([0, 1, 1, 1, 1]),
top_k=5,
separate_camera_set=False,
single_gallery_shot=False)
assert np.all(result == [0.6, 0.6, 0.6, 1, 1])
def test_only_distance_matrix(self):
distance_matrix = np.array([[0, 1, 2, 3, 4], [1, 0, 2, 3, 4],
[0, 1, 2, 3, 4], [0, 1, 2, 3, 4],
[1, 2, 3, 4, 0]])
m, n = distance_matrix.shape
result = eval_cmc(distance_matrix,
query_ids=np.arange(m),
gallery_ids=np.arange(n),
query_cams=np.zeros(m).astype(np.int32),
gallery_cams=np.ones(n).astype(np.int32))
assert np.all(result[:5] == [0.6, 0.6, 0.8, 1.0, 1.0])
def test_duplicate_ids(self):
distance_matrix = np.array([[0, 1, 2, 3], [0, 1, 2, 3], [0, 1, 2, 3],
[0, 1, 2, 3]])
result = eval_cmc(
distance_matrix,
query_ids=np.array([0, 0, 1, 1]),
gallery_ids=np.array([0, 0, 1, 1]),
top_k=4,
gallery_cams=np.ones(distance_matrix.shape[1]).astype(np.int32),
query_cams=np.zeros(distance_matrix.shape[0]).astype(np.int32),
separate_camera_set=False,
single_gallery_shot=False)
assert np.all(result == [0.5, 0.5, 1, 1])