aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-05 18:46:50 +0000
committernunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-05 18:46:50 +0000
commit48ac66ef7f26a4d503e5850af1ee2ad345fd6dc7 (patch)
treeb37061ef38934d4afec76d193c71d227d3c78685
parent769a50e70ac253531229e1639db6bc9e401a0c43 (diff)
downloadvz215_np1915-48ac66ef7f26a4d503e5850af1ee2ad345fd6dc7.tar.gz
vz215_np1915-48ac66ef7f26a4d503e5850af1ee2ad345fd6dc7.tar.bz2
vz215_np1915-48ac66ef7f26a4d503e5850af1ee2ad345fd6dc7.zip
Fix lambda rerank
-rwxr-xr-xpart2.py4
1 files 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')