OpenGlue icon indicating copy to clipboard operation
OpenGlue copied to clipboard

The weight of the SuperPoint

Open Angelababypro opened this issue 2 years ago • 2 comments

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

Angelababypro avatar Nov 29 '22 04:11 Angelababypro

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

ducha-aiki avatar Nov 29 '22 07:11 ducha-aiki

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.

Angelababypro avatar Nov 29 '22 11:11 Angelababypro