search

jono-m / OrganoID

22 stars 12 forks source link

Any script available to train the model on a custom dataset? #9

Open ajinkya-kulkarni opened 1 year ago

ajinkya-kulkarni commented 1 year ago

Hello team, thanks for the repo. Is there any script/notebook available to train the model on a custom dataset? Thanks!

schmoogol commented 1 year ago

Have a look at the Split.py, Augment.py and Train.py files in /CommandLine. These are what you can use to prepare a dataset and train your own model (e.g. python OrganoID.py split/augment/train...).

ajinkya-kulkarni commented 1 year ago

Ah okay, thanks! I already have split my dataset and Augmented it from my end, so I think Train.py will be the most helpful to me. And as I understand the code, Train.py requires the training data as images in images and the corresponding masks in segmentations right?

schmoogol commented 1 year ago

Correct, and you also need a validation dataset (/validation/images and /validation/segmentations).