bob.ip.binseg.models.driu_pix

Functions

driu_pix([pretrained_backbone, progress])

Builds DRIU with pixelshuffle by adding backbone and head together

Classes

DRIUPIX([in_channels_list])

DRIUPIX head module.

class bob.ip.binseg.models.driu_pix.DRIUPIX(in_channels_list=None)

Bases: Module

DRIUPIX head module. DRIU with pixelshuffle instead of ConvTrans2D

Parameters

in_channels_list (list) – number of channels for each feature map that is returned from backbone

forward(x)

Parameters

x (list) – list of tensors as returned from the backbone network. First element: height and width of input image. Remaining elements: feature maps for each feature level.

Return type

torch.Tensor

bob.ip.binseg.models.driu_pix.driu_pix(pretrained_backbone=True, progress=True)

Builds DRIU with pixelshuffle by adding backbone and head together

Parameters

• pretrained_backbone (bool, Optional) – If set to True, then loads a pre-trained version of the backbone (not the head) for the DRIU network using VGG-16 trained for ImageNet classification.

• progress (bool, Optional) – If set to True, and you decided to use a pretrained_backbone, then, shows a progress bar of the backbone model downloading if download is necesssary.

Returns

module – Network model for DRIU (vessel segmentation) with pixelshuffle

Return type

torch.nn.Module