segmentation_models.pytorch
segmentation_models.pytorch copied to clipboard
Using "segmentation_models_pytorch" with Unet++
was trying to use segmentation_models_pytorch with Unet++. I tried to look at the example at HuBMAP - Pytorch smp Unet++. My understanding Unet++ returns 4 outputs (as part of deep supervision). But in the example, I notice it is only returning one output. I think I'm missing something here.