Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Using your trained model #12

Closed
Amir-Reza-Sadri opened this issue Nov 18, 2020 · 5 comments
Closed

Using your trained model #12

Amir-Reza-Sadri opened this issue Nov 18, 2020 · 5 comments

Comments

@Amir-Reza-Sadri
Copy link

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?

@thomaskuestner
Copy link
Member

Which pre-trained model do you want to use?

@Amir-Reza-Sadri
Copy link
Author

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?

@thomaskuestner
Copy link
Member

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.

@Amir-Reza-Sadri
Copy link
Author

Amir-Reza-Sadri commented Nov 30, 2020 via email

@thomaskuestner
Copy link
Member

Solved.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants