yingxin-jia / SuperGlue-pytorch

[SuperGlue: Learning Feature Matching with Graph Neural Networks] This repo includes PyTorch code for training the SuperGlue matching network on top of SIFT keypoints and descriptors.
Other
545 stars 126 forks source link

Must training with COCO2014 dataset? #25

Open brfIMU opened 2 years ago

brfIMU commented 2 years ago

Can I use my dataset to train the model? (It's always wrong when I use ma dataset) Why there is a mistake as soon as I change the batch-size? Looking forward to your reply! Thank you!

sunset326 commented 1 year ago

batch-size cant modify,must be 1