From aaee9b10aab7c78c7a02d5caec4ed37b85dff1bd Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Sat, 8 Dec 2018 17:37:25 +0000 Subject: Use correct transpose for mahalanobis --- evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/evaluate.py b/evaluate.py index e8468a2..38b20bb 100755 --- a/evaluate.py +++ b/evaluate.py @@ -84,7 +84,7 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam else: if args.mahalanobis: # metric = 'jaccard' is also valid - cov_inv = np.linalg.inv(np.cov(gallery_data.T)) + cov_inv = np.linalg.inv(np.cov(gallery_data.T)).T distances = cdist(probe_data, gallery_data, 'mahalanobis', VI=cov_inv) else: distances = cdist(probe_data, gallery_data, 'euclidean') -- cgit v1.2.3-54-g00ecf