mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
fix evaluator
This commit is contained in:
parent
3880b23ab4
commit
1db050b51f
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user