Using pretained model in our code

[sart-leo]

Hi, I was wondering how to easily use a pretained model (Task005_Prostate) in the code in an easy way. I would like to extract some feature maps of intermediate layers and reuse the latent representation of the bottleneck of one of UNet. I am not very familiar with the framework and you could be of great help.

[FabianIsensee]

You should just follow the code of nnUNet_predict. You can replicate it in your script and then access the model. In order to gett access to the feature maps you will need to modify th Generic_UNet