From 1db050b51f4e722e76e2fcba976255647ae67c0b Mon Sep 17 00:00:00 2001 From: "zuchen.wang" Date: Wed, 27 Oct 2021 16:15:37 +0800 Subject: [PATCH] fix evaluator --- fastreid/evaluation/pair_score_evaluator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastreid/evaluation/pair_score_evaluator.py b/fastreid/evaluation/pair_score_evaluator.py index ddf032e..2b8d206 100644 --- a/fastreid/evaluation/pair_score_evaluator.py +++ b/fastreid/evaluation/pair_score_evaluator.py @@ -18,7 +18,7 @@ class PairScoreEvaluator(PairEvaluator): def process(self, inputs, outputs): prediction = { - 'distances': outputs['cls_outputs'].to(self._cpu_device).numpy(), - 'labels': inputs["targets"].to(self._cpu_device).numpy() + 'distances': outputs['cls_outputs'][:, 1].to(self._cpu_device).numpy(), + 'labels': inputs["targets"].to(self._cpu_device).numpy() } self._predictions.append(prediction)