search

mihaidusmanu / d2-net

D2-Net: A Trainable CNN for Joint Description and Detection of Local Features
Other
761 stars 163 forks source link

Training on custom datasets #81

Closed sachitha-bandara closed 3 years ago

sachitha-bandara commented 3 years ago

Hi, could you explain how to train the model with a custom dataset?

mihaidusmanu commented 3 years ago

Hello. The current implementation only supports training on MegaDepth. You will have to implement your own dataloader to train on a different dataset. Please refer to the following issue for more details: https://github.com/mihaidusmanu/d2-net/issues/60.