niftynet.layer.base_layer module

class Invertible

Bases: object

interface of Invertible data

inverse_op(*args, **kwargs)

class Layer(name='untitled_op')

Bases: object

layer_op(*args, **kwargs)

layer_scope()

to_string()

class TrainableLayer(name='trainable_op')

Bases: niftynet.layer.base_layer.Layer

Extends the Layer object to have trainable parameters, adding initializers and regularizers.

trainable_variables()

restore_from_checkpoint(checkpoint_name, scope=None)

regularizer_loss()

num_trainable_params()

to_string()

initializers

regularizers

class DataDependentLayer(name='data_dependent_op')

Bases: niftynet.layer.base_layer.Layer

Some layers require a one-pass training through the training set to determine their internal models, this abstract provides interfaces for training these internal models and querying the status.

is_ready()

train(*args, **kwargs)

class RandomisedLayer(name='randomised_op')

Bases: niftynet.layer.base_layer.Layer

The layers require a randomisation process, to randomly change some of the layer’s states on the fly.

randomise(*args, **kwargs)

class LayerFromCallable(layer_op, name='from_callable_op')

Bases: niftynet.layer.base_layer.Layer

Module wrapping a function provided by the user. Analogous to snt.Module

layer_op(*args, **kwargs)