3d classification with multi output head

[YingdiZhang]

Thanks for the great job. I am using monai in a 3D classification task. I want to classify a CT scan in two ways, the first is binary classification of if the disease exist, the second is classify another disease into not exist, low level or high level. I want to achieve this classification task via multi output head, how to achieve that?

[KumoLiu]

Hi @YingdiZhang,

You can create your own custom network that includes multiple heads.

Here is an example in MONAI which outputs three heads. You can refer to it.

MONAI/monai/networks/nets/hovernet.py

Line 407 in 2daabf9

class HoVerNet(nn.Module):

Hope it helps, thanks!