Spaces:

Realcat
/

image-matching-webui

Running

Realcat commited on May 8

Commit

5ff702b

1 Parent(s): 801bbff

fix: select kpts

Files changed (1) hide show

imcui/hloc/extractors/liftfeat.py CHANGED Viewed

@@ -13,7 +13,7 @@ from models.liftfeat_wrapper import LiftFeat, MODEL_PATH
 def select_idx(N, M):
-    numbers = list(range(1, N + 1))
     selected = random.sample(numbers, M)
     return selected

 def select_idx(N, M):
+    numbers = list(range(0, N))
     selected = random.sample(numbers, M)
     return selected