mednet.models.segment

Specialized semantic segmentation models and associated functionality.

Modules

backbones

driu

DRIU network architecture for vessel segmentation, from [MPTAVG16].

driu_bn

DRIU network architecture, with added batch-normalization.

driu_od

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

driu_pix

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

hed

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

lwnet

Little W-Net (LWNET) network architecture, from [GAD+22].

m2unet

Mobile2 UNet network architecture, from [LWJ18].

make_layers

model

Definition of base model type for semantic segmentation tasks.

unet

UNet network architecture, from [RFB15].