Spaces:
Running
Running
fix: select kpts
Browse files
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(
|
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 |
|