diff --git a/README.md b/README.md index d053645..b4d7279 100644 --- 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 index 6639b62..588ecc9 100644 --- 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: