From 579a5cf552426070e589564f760d2d84ad432fb7 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Mon, 18 May 2020 17:05:20 +0800 Subject: [PATCH] fix: modify re-rank typro --- fastreid/evaluation/reid_evaluation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastreid/evaluation/reid_evaluation.py b/fastreid/evaluation/reid_evaluation.py index 82d88c5..8686de5 100644 --- a/fastreid/evaluation/reid_evaluation.py +++ b/fastreid/evaluation/reid_evaluation.py @@ -83,7 +83,7 @@ class ReidEvaluator(DatasetEvaluator): if self.cfg.TEST.RERANK.ENABLED: logger.info("Test with rerank setting") k1 = self.cfg.TEST.RERANK.K1 - k2 = self.cfg.TEST.RERANK.K1 + k2 = self.cfg.TEST.RERANK.K2 lambda_value = self.cfg.TEST.RERANK.LAMBDA q_q_dist = self.cal_dist(self.cfg.TEST.METRIC, query_features, query_features) g_g_dist = self.cal_dist(self.cfg.TEST.METRIC, gallery_features, gallery_features)