Realcat commited on
Commit
5ff702b
·
1 Parent(s): 801bbff

fix: select kpts

Browse files
Files changed (1) hide show
  1. imcui/hloc/extractors/liftfeat.py +1 -1
imcui/hloc/extractors/liftfeat.py CHANGED
@@ -13,7 +13,7 @@ from models.liftfeat_wrapper import LiftFeat, MODEL_PATH
13
 
14
 
15
  def select_idx(N, M):
16
- numbers = list(range(1, N + 1))
17
  selected = random.sample(numbers, M)
18
  return selected
19
 
 
13
 
14
 
15
  def select_idx(N, M):
16
+ numbers = list(range(0, N))
17
  selected = random.sample(numbers, M)
18
  return selected
19