Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The weight of the SuperPoint #10

Open
Angelababypro opened this issue Nov 29, 2022 · 2 comments
Open

The weight of the SuperPoint #10

Angelababypro opened this issue Nov 29, 2022 · 2 comments

Comments

@Angelababypro
Copy link

Angelababypro commented Nov 29, 2022

the superpoint weight you provided is not match with the name in both superpoint_coco.yaml and superpoint_kitti.yaml. And use the weight you provided for superpoint will raise the keyerror as follow:
File "d:\Develop\PythonProject\OpenGlue-main\models\features\superpoint\model.py", line 175, in _load_weights
state_dict = torch.load(str(weights), map_location='cpu')['model_state_dict']
KeyError: 'model_state_dict'

I can not train superpoint right now so could you please provide the weight for superpoint-coco or superpoint-kitti?

The first line directory in the screenshot is the default checkpoint path and the second line is the checkpoint path direct to the weight that you give. and it will cause the Keyerror.
Snipaste_2022-11-29_12-43-16

@ducha-aiki
Copy link
Collaborator

https://github.com/eric-yyjau/pytorch-superpoint/tree/master/logs

@Angelababypro
Copy link
Author

https://github.com/eric-yyjau/pytorch-superpoint/tree/master/logs

you have no idea how much that I thank you, wish you will not encounter any bugs in the whole year in the future.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants