Skip to content

The weight of the SuperPoint #10

@Angelababypro

Description

@Angelababypro

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions