deepdraw.models.driu_pix#
Functions
|
Builds DRIU with pixelshuffle by adding backbone and head together. |
Classes
|
DRIUPIX head module. |
- class deepdraw.models.driu_pix.DRIUPIX(in_channels_list=None)[source]#
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
- deepdraw.models.driu_pix.driu_pix(pretrained_backbone=True, progress=True)[source]#
Builds DRIU with pixelshuffle by adding backbone and head together.
- Parameters:
pretrained_backbone (
bool
, Optional) – If set toTrue
, 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 toTrue
, and you decided to use apretrained_backbone
, then, shows a progress bar of the backbone model downloading if download is necesssary.
- Returns:
module (
torch.nn.Module
) – Network model for DRIU (vessel segmentation) with pixelshuffle