Vincentqyw
commited on
Commit
•
e0f1139
1
Parent(s):
6f58dfa
update: lanet
Browse files- hloc/extractors/lanet.py +3 -3
hloc/extractors/lanet.py
CHANGED
@@ -53,7 +53,7 @@ class LANet(BaseModel):
|
|
53 |
scores = scores[idxs]
|
54 |
|
55 |
return {
|
56 |
-
"keypoints":
|
57 |
-
"scores":
|
58 |
-
"descriptors":
|
59 |
}
|
|
|
53 |
scores = scores[idxs]
|
54 |
|
55 |
return {
|
56 |
+
"keypoints": keypoints[None],
|
57 |
+
"scores": scores[None],
|
58 |
+
"descriptors": descriptors.T[None],
|
59 |
}
|