niftynet.network.vnet module

class VNet(num_classes, w_initializer=None, w_regularizer=None, b_initializer=None, b_regularizer=None, acti_func='relu', name='VNet')

Bases: niftynet.network.base_net.BaseNet

implementation of V-Net:

Milletari et al., “V-Net: Fully convolutional neural networks for volumetric medical image segmentation”, 3DV ‘16

layer_op(images, is_training, layer_id=-1)

class VNetBlock(func, n_conv, n_feature_chns, n_output_chns, w_initializer=None, w_regularizer=None, b_initializer=None, b_regularizer=None, acti_func='relu', name='vnet_block')

Bases: niftynet.layer.base_layer.TrainableLayer

layer_op(main_flow, bypass_flow)