Update URLs (#2)

cvg · Jun 26, 2023 · e9e997d · e9e997d
1 parent a4c268f
commit e9e997d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -87,7 +87,6 @@ If you use any ideas from the paper or code from this repo, please consider citi
                Marc Pollefeys},
   title     = {{LightGlue}: Local Feature Matching at Light Speed},
   booktitle = {ArXiv PrePrint},
-  year      = {2023},
-  url       = {psarlin.com/lightglue.pdf}
+  year      = {2023}
 }
 ```
diff --git a/lightglue/superpoint.py b/lightglue/superpoint.py
@@ -134,8 +134,8 @@ def __init__(self, **conf):
             c5, self.config['descriptor_dim'],
             kernel_size=1, stride=1, padding=0)
 
-        path = Path(__file__).parent / 'weights/superpoint_v1.pth'
-        self.load_state_dict(torch.load(str(path)))
+        url = "https://github.com/cvg/LightGlue/releases/download/v0.1_arxiv/superpoint_v1.pth"
+        self.load_state_dict(torch.hub.load_state_dict_from_url(url))
 
         mk = self.config['max_num_keypoints']
         if mk == 0 or mk < -1: