search

feixue94 / imp-release

[CVPR 2023] IMP: iterative matching and pose estimation with transformer-based recurrent module
https://feixue94.github.io/
Other
141 stars 11 forks source link

Train with my own dataset #8

Closed Strollercarra closed 1 year ago

Strollercarra commented 1 year ago

Dear author, thanks a lot for your released codes. I am wondering that how to train with my own dataset in the codes. Would you mind giving me some instances in your codes? Looking forward to your early reply.

feixue94 commented 1 year ago

Hi, Only keypoints (descriptors, scores, coordinates) and gt-matches of image pairs are needed for training. If you want to train the imp on your own dataset, you can follow the code in dump/dump_megadepth.py to build your own training samples and use the training script to start the training.

feixue94 commented 1 year ago

This issue is closed. If you have any other questions, feel free to reopen it.