From 48ac66ef7f26a4d503e5850af1ee2ad345fd6dc7 Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 5 Dec 2018 18:46:50 +0000 Subject: Fix lambda rerank --- part2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/part2.py b/part2.py index 82a8cdd..08b27ac 100755 --- a/part2.py +++ b/part2.py @@ -74,12 +74,12 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam if args.rerank: distances = re_ranking(probe_data, gallery_data, - args.reranka ,args.rerankb , 0.3, + args.reranka, args.rerankb, args.rerankl, MemorySave = False, Minibatch = 2000) else: if args.mahalanobis: # metric = 'jaccard' is also valid - distances = cdist(probe_data, gallery_data, 'sqeuclidean') + distances = cdist(probe_data, gallery_data, 'jaccard') else: distances = cdist(probe_data, gallery_data, 'euclidean') -- cgit v1.2.3-54-g00ecf