MonAI Unet -> linear output #8295
System-Lord22
started this conversation in
General
Replies: 2 comments
-
The appropriateness of this depends on your task. To get at the middle/bottom layer of a UNet, here is one approach: Project-MONAI/tutorials#949 |
Beta Was this translation helpful? Give feedback.
0 replies
-
Can I get access to the UNet forward function from somewhere so I can subclass and override it? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I've been using a MinAI UNet to examine an image and then an AdaptiveAvgPool3d to put it into a number of channels to go into a fully connected layer and then give a linear output (a "score").
It seems to work alright but someone said it was more traditional to just capture the middle layer of the UNet as the input to the FC layer.
Any opinions? And how would I easily chop the UNet in half like that?
Thanks
Beta Was this translation helpful? Give feedback.
All reactions