Vincentqyw commited on
Commit
e0f1139
1 Parent(s): 6f58dfa

update: lanet

Browse files
Files changed (1) hide show
  1. 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": torch.from_numpy(keypoints)[None],
57
- "scores": torch.from_numpy(scores)[None],
58
- "descriptors": torch.from_numpy(descriptors.T)[None],
59
  }
 
53
  scores = scores[idxs]
54
 
55
  return {
56
+ "keypoints": keypoints[None],
57
+ "scores": scores[None],
58
+ "descriptors": descriptors.T[None],
59
  }