mednet.models.segmentΒΆ

Specialized semantic segmentation models and associated functionality.

Modules

backbones

driu

DRIU network architecture for vessel segmentation, from [MANINIS-2016].

driu_bn

DRIU network architecture, with added batch-normalization.

driu_od

DRIU network architecture for optic-disc segmentation, from [MANINIS-2016].

driu_pix

DRIU network architecture, with torch.nn.PixelShuffle instead of ConvTrans2D.

hed

Holistically-Nested Edge Detection (HED) network architecture, from [XIE-2015].

lwnet

Little W-Net (LWNET) network architecture, from [GALDRAN-2020].

m2unet

Mobile2 UNet network architecture, from [LAIBACHER-2018].

make_layers

model

Definition of base model type for semantic segmentation tasks.

unet

UNet network architecture, from [RONNEBERGER-2015].