search

ANTsX / ANTsPyNet

Pre-trained models and utilities for deep learning on medical images in Python
https://antspynet.readthedocs.io
Apache License 2.0
201 stars 29 forks source link

Any example of training model with my own dataset? #140

Closed kingaza closed 1 month ago

kingaza commented 1 month ago

Hello guys,

It is a great job. Now I want to use it with my own dataset, however, I cannot find any snippet about training.

thanks.

Best Regards, kingaza

ntustison commented 1 month ago

Training is not something we formally support.

However, all the components to perform training are available in the ANTsX repositories. I also post all the training scripts I use for the distributed models here.

kingaza commented 1 month ago

Got it. thx.