search

rpautrat / SuperPoint

Efficient neural feature detector and descriptor
MIT License
1.91k stars 418 forks source link

. pth file inference HPatches #329

Open haoxintian opened 4 hours ago

haoxintian commented 4 hours ago

Hello, I would like to ask you about how to use the official pre trained weight file to evaluate the repeatability of HPatches inference?

I have completed the training of Coco using superpoint, but I found that export_detection.py does not support the push of. pth weight files. How can I solve this problem?

rpautrat commented 3 hours ago

Hi, the .pth model is a model that has been converted to Pytorch. But the evaluation on HPatches is done on the original Tensorflow model (the one you obtained after training on COCO). If you want to compare to the one we trained, you can download the Tensorflow model from the links in the ReadMe: https://github.com/rpautrat/SuperPoint?tab=readme-ov-file#fine-tuning-a-pretrained-model

haoxintian commented 3 hours ago

I understand, thank you very much for your answer!