niftynet.network.unet module

class 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=True, layer_id=-1, **unused_kwargs)

class 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)