3d dense UNET for vertebrae segmentation about 3d-brain-segmentation HOT 1 CLOSED

Hi,

well there are multiple possible routes how to proceed, but If you want to implement the vertebrae segmentation and classification as an end-to-end task using single neural network, then I would treat this problem as a multi class segmentation.

You would have to encode the data properly and then change the output layer of the net to softmax activation with as many channels as there are classes. This article describes something similar:
https://naomi-fridman.medium.com/multi-class-image-segmentation-a5cc671e647a

Hope this helps :)

Take care,

Martin

from 3d-brain-segmentation.