Update evaluate_rerank_gpu.py

layumi · Aug 17, 2023 · ad62bfe · ad62bfe
1 parent aea3172
commit ad62bfe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GPU-Re-Ranking/evaluate_rerank_gpu.py b/GPU-Re-Ranking/evaluate_rerank_gpu.py
@@ -38,6 +38,7 @@
                     default=7,      # Market-1501
                     # default=10,   # Veri-776
                     help='parameter k2')
+parser.add_argument('--lamb', type=float, default=0.3)
 
 args = parser.parse_args()
 
@@ -54,7 +55,7 @@ def main():
     query_feature = query_feature.cuda()
     gallery_feature = gallery_feature.cuda()
 
-    indices = gnn_reranking(query_feature, gallery_feature, args.k1, args.k2)
+    indices = gnn_reranking(query_feature, gallery_feature, args.k1, args.k2, args.lamb)
     evaluate_ranking_list(indices, query_label, query_cam, gallery_label, gallery_cam)
 
 if __name__ == '__main__':