mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix evaluator

pull/608/head
zuchen.wang 2021-10-27 16:15:37 +08:00
parent 3880b23ab4
commit 1db050b51f
1 changed files with 2 additions and 2 deletions
fastreid/evaluation

4
fastreid/evaluation/pair_score_evaluator.py
View File

@ -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)