lab-midas/med_segmentation

Using your trained model

Amir-Reza-Sadri opened this issue · 5 comments

I would like to use your trained model over my dataset. Would it be possible to use your DL model as a pre-trained model and do a transfer learning?

Which pre-trained model do you want to use?

The one for the whole-body adipose tissue segmentation. If possible the weights and network specifications on both TUEF/DZD and NAKO dataset. May I use them in PyTorch?

You can find the network specifications in the config files pretrained_model/AT_MRGEN_New and pretrained_model/AT_NAKO.
All the networks were trained on this code using Tensorflow, which can not easily be used in Pytorch. If you want to use it in Pytorch you would first need to check that all functions (and parametrizations) are implemented there exactly the same. Please send me an email if you are still interested in the weights of the network.

Solved.