magicleap / SuperPointPretrainedNetwork

PyTorch pre-trained model for real-time interest point detection, description, and sparse tracking (https://arxiv.org/abs/1712.07629)
Other
1.93k stars 397 forks source link

train custum superpoint or superglue #15

Open roboticsai opened 3 years ago

roboticsai commented 3 years ago

How do i train my custum superpint and superglue models? Any git repo or package? Instead of using already available dataset?

shaofengzeng commented 3 years ago

Here is a pytorch version that has implemented most of the related functions https://github.com/shaofengzeng/MySuperPoint You can finish your work based on this code