lint

cvg · Jan 23, 2024 · cd6ab70 · cd6ab70
1 parent 0f413c2
commit cd6ab70
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lightglue/dog_hardnet.py b/lightglue/dog_hardnet.py
@@ -12,7 +12,6 @@ def __init__(self, **conf):
         super().__init__(**conf)
         self.laf_desc = LAFDescriptor(HardNet(True)).eval()
 
-
     def forward(self, data: dict) -> dict:
         image = data["image"]
         if image.shape[1] == 3:
@@ -39,4 +38,4 @@ def forward(self, data: dict) -> dict:
             p["descriptors"] = self.laf_desc(img[None], lafs).reshape(-1, 128)
             pred.append(p)
         pred = {k: torch.stack([p[k] for p in pred], 0).to(device) for k in pred[0]}
-        return pred
+        return pred