Update evaluators.py

ShengcaiLiao · Mar 31, 2021 · e057350 · e057350
1 parent 2763bf5
commit e057350
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/reid/evaluators.py b/reid/evaluators.py
@@ -202,9 +202,9 @@ def evaluate(self, matcher, testset, query_loader, gallery_loader, gal_batch_siz
             with torch.no_grad():
                 dist_rerank[:num_prob, num_prob:] = dist
                 dist_rerank[num_prob:, :num_prob] = dist.t()
-                dist_rerank[:num_prob, :num_prob] = pairwise_distance(prob_fea, prob_fea, matcher, gal_batch_size,
+                dist_rerank[:num_prob, :num_prob] = pairwise_distance(matcher, prob_fea, prob_fea, gal_batch_size,
                                                                     prob_batch_size)
-                dist_rerank[num_prob:, num_prob:] = pairwise_distance(gal_fea, gal_fea, matcher, gal_batch_size,
+                dist_rerank[num_prob:, num_prob:] = pairwise_distance(matcher, gal_fea, gal_fea, gal_batch_size,
                                                                     prob_batch_size)
 
             dist_rerank = reranking(dist_rerank, num_prob, verbose=True)