From c833d679406ff6520fcd6ea41f115f48f084ccb0 Mon Sep 17 00:00:00 2001 From: xukuanHIT <403282859@qq.com> Date: Sat, 4 Dec 2021 07:05:50 +0800 Subject: [PATCH] Update offline_topK.py --- experiments/place_recogination/offline_topK.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/experiments/place_recogination/offline_topK.py b/experiments/place_recogination/offline_topK.py index cebea96..c40d76f 100644 --- a/experiments/place_recogination/offline_topK.py +++ b/experiments/place_recogination/offline_topK.py @@ -160,8 +160,14 @@ def relocalization_offline(configs): if k > len(scores): k_loop_images = match_images else: - _, indices = scores.topk(k) - indices = indices.numpy().tolist() + value, _ = scores.topk(k) + min_v = value[-1].item() + indices = [] + for i_score in range(len(scores)): + if(scores[i_score] >= min_v): + indices.append(i_score) + # _, indices = scores.topk(k) + # indices = indices.numpy().tolist() for idx in indices: k_loop_images.append(match_images[idx])