niftynet.network.resnet module¶

class ResNetDesc(bn, fc, conv1, blocks)¶

Bases: tuple

blocks¶: Alias for field number 3

bn¶: Alias for field number 0

conv1¶: Alias for field number 2

fc¶: Alias for field number 1

class ResNet(num_classes, n_features=[16, 64, 128, 256], n_blocks_per_resolution=10, w_initializer=None, w_regularizer=None, b_initializer=None, b_regularizer=None, acti_func='relu', name='ResNet')[source]¶

Bases: niftynet.network.base_net.BaseNet

implementation of Res-Net:: He et al., “Identity Mappings in Deep Residual Networks”, arXiv:1603.05027v3

create()[source]¶

layer_op(images, is_training=True, **unused_kwargs)[source]¶

class BottleneckBlockDesc1(conv)¶

Bases: tuple

conv¶: Alias for field number 0

class BottleneckBlockDesc2(common_bn, conv, conv_shortcut)¶

Bases: tuple

common_bn¶: Alias for field number 0

conv¶: Alias for field number 1

conv_shortcut¶: Alias for field number 2

class BottleneckBlock(n_output_chns, stride, Conv, name='bottleneck')[source]¶

Bases: niftynet.layer.base_layer.TrainableLayer

create(input_chns)[source]¶

layer_op(images, is_training=True)[source]¶

class DownResBlockDesc(blocks)¶

Bases: tuple

blocks¶: Alias for field number 0

class DownResBlock(n_output_chns, count, stride, Conv, name='downres')[source]¶

Bases: niftynet.layer.base_layer.TrainableLayer

create()[source]¶

layer_op(images, is_training)[source]¶