niftynet.network.unet module¶
-
class
niftynet.network.unet.
UNet3D
(num_classes, w_initializer=None, w_regularizer=None, b_initializer=None, b_regularizer=None, acti_func='prelu', name='UNet')¶ Bases:
niftynet.layer.base_layer.TrainableLayer
- reimplementation of 3D U-net
- Çiçek et al., “3D U-Net: Learning dense Volumetric segmentation from sparse annotation”, MICCAI ‘16
-
layer_op
(images, is_training, layer_id=-1)¶
-
class
niftynet.network.unet.
UNetBlock
(func, n_chns, kernels, w_initializer=None, w_regularizer=None, with_downsample_branch=False, acti_func='relu', name='UNet_block')¶ Bases:
niftynet.layer.base_layer.TrainableLayer
-
layer_op
(input_tensor, is_training)¶
-