cvg / glue-factory

Training library for local feature detection and matching
Apache License 2.0
771 stars 99 forks source link

Training on Custom Dataset #113

Open Shafaq-S opened 2 weeks ago

Shafaq-S commented 2 weeks ago

I am training LightGlue on my own dataset (which consists of RGB images only) by running this experiment: python -m gluefactory.train sp+lg_homography --conf gluefactory/configs/superpoint-open+lightglue_homography.yaml However, when I run inference after training, I get no matches at all. Any leads to solve this problem? Also, is it possible to visualize matches during training after each epoch?