mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

fix evaluator

Browse Source
This commit is contained in:
zuchen.wang 2021-10-27 16:15:37 +08:00
parent 3880b23ab4
commit 1db050b51f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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)