attempt to use the encoder part of UNode to classify?

DIAGNijmegen / neural-odes-segmentation

Neural Ordinary Differential Equations for Semantic Segmentation of Individual Colon Glands

MIT License

106 stars 28 forks source link

attempt to use the encoder part of UNode to classify? #11

Open saysth opened 3 years ago

saysth commented 3 years ago

hey guys, since the encoder part of the UNet can be modified to a great classifier, I also used the encoder part of your UNode on a simple 2-category classification task, but the performances were very unstable while UNet-encoder reached 90%+ stably.
Could you provide any suggestions on this problem? Very appreciated!

hanspinckaers commented 3 years ago

Hi @saysth, sounds interesting, could you share some code?