Spaces:
Running on Zero
Running on Zero
fix: xfeat_lightglue
Browse files
hloc/matchers/xfeat_lightglue.py
CHANGED
|
@@ -37,7 +37,11 @@ class XFeatLightGlue(BaseModel):
|
|
| 37 |
]
|
| 38 |
out0.update({"image_size": (im0.shape[-1], im0.shape[-2])}) # W H
|
| 39 |
out1.update({"image_size": (im1.shape[-1], im1.shape[-2])}) # W H
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
mkpts_0 = torch.from_numpy(mkpts_0) # n x 2
|
| 42 |
mkpts_1 = torch.from_numpy(mkpts_1) # n x 2
|
| 43 |
pred = {
|
|
|
|
| 37 |
]
|
| 38 |
out0.update({"image_size": (im0.shape[-1], im0.shape[-2])}) # W H
|
| 39 |
out1.update({"image_size": (im1.shape[-1], im1.shape[-2])}) # W H
|
| 40 |
+
pred = self.net.match_lighterglue(out0, out1)
|
| 41 |
+
if len(pred) == 3:
|
| 42 |
+
mkpts_0, mkpts_1, _ = pred
|
| 43 |
+
else:
|
| 44 |
+
mkpts_0, mkpts_1 = pred
|
| 45 |
mkpts_0 = torch.from_numpy(mkpts_0) # n x 2
|
| 46 |
mkpts_1 = torch.from_numpy(mkpts_1) # n x 2
|
| 47 |
pred = {
|