Training on my own dataset

[CalleRC]

Hi,

Is there any way to import your own dataset to finetune the already pretrained superpoint+lightglue model?

I have looked through the code and from my understanding you only support finetuning on the megadepth dataset but not any custom created dataset for finetuning.

I have created my own dataset following the structure from megadepth1500. Do I need any other files? I Can see that you are importing the depthmap but is that necessary when using the camera translation, rotation and intrinsics? Also if I need other files than images and pairs_calibrated what files are needed in order to train using my own dataset (in the case that this is possible using glue-factory).

I wish to finetune on my own dataset for two reasons.

1. I do not have enough space to download a 400gb dataset
2. Finetuning for my special use case may result in a better feature matcher

[tenngre]

Can you train on own dataset now?

[MrPolo-AI]

Hi, do you find a way to train your own dataset now?

[heitieya]

Hello, may I ask if you know how to implement your own dataset for training? Thank you very much

[vampire1204]

1. rewrite the API of your datasets
2. select right file of config (yaml) or change it

[huangchenxin]

Hello, Can you train on own dataset now? And may I ask if you know how to create your own dataset? Implement by COLMAP? Thank you very much