diff options
Diffstat (limited to 'evaluate.py')
| -rwxr-xr-x | evaluate.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/evaluate.py b/evaluate.py index e90dae8..c5528af 100755 --- a/evaluate.py +++ b/evaluate.py @@ -76,13 +76,13 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam debug("gallery shape: %s", gallery_data.shape) if args.rerank: - distances = re_ranking(probe_data, gallery_data, - args.reranka ,args.rerankb , 0.3, + distances = re_ranking(probe_data, gallery_data, + 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') |