search

MIDA-group / CoMIR

Framework for joint representation learning, evaluation through multimodal registration and comparison with image translation based approaches
Other
74 stars 11 forks source link

How to train dataset? #52

Closed Jane-HZ closed 2 years ago

Jane-HZ commented 2 years ago

Thanks for your great contribution! There is a problem: if I want to train my own dataset, what should I do? Your kind help will be much appreciated.

npielawski commented 2 years ago

Hello! Thanks for your interest!

I suggest you set up a training and testing datasets containing aligned images, not many are necessary, so manual registration is an option. Then you can take a look at the following notebook: https://github.com/MIDA-group/CoMIR/blob/master/train-biodata.ipynb which contains all you need to train and test your model.

Feel free to ask if you have any issues trying out our method.

Jane-HZ commented 2 years ago

Thank you very much for your patient reply!