search

wasserth / TotalSegmentator

Tool for robust segmentation of >100 important anatomical structures in CT and MR images
Apache License 2.0
1.51k stars 250 forks source link

Do you have the model architecture anywhere? #165

Closed ZanChaudhry closed 1 year ago

ZanChaudhry commented 1 year ago

Hello! Thank you for making such a great segmentation toolkit. I was interested in using the pre-trained weights of the organ segmentation pipeline for a transfer learning problem; however, for this I would need the model architecture, not just the weights. The pre-trained weights are in a .MODEL file, which seems to contain a state_dict with the weights, as well as some other information related to model training, but not the architecture itself. Is the model architecture somewhere in this repository?

wasserth commented 1 year ago

The model is normal nnunet model. You can check the nnunet code + documentation (https://github.com/MIC-DKFZ/nnUNet) for some more information. I would have to search for the model architecture myself.