InstanceNormalizationLayer: Creates an instance normalization layer

In ANTsX/ANTsRNet: Neural Networks for Medical Image Processing

Creates an instance normalization layer

Description

Creates an instance normalization layer as described in the paper

Details

https://arxiv.org/abs/1701.02096

with the implementation ported from the following python implementation

https://github.com/keras-team/keras-contrib/blob/master/keras_contrib/layers/normalization/instancenormalization.py

Value

an instance normalization layer

Arguments

axis

Integer specifying which axis should be normalized, typically the feature axis. For example, after a Conv2D layer with channels_first, set axis = 2. Setting axis=-1L will normalize all values in each instance of the batch. Axis 1 is the batch dimension for tensorflow backend so we throw an error if axis = 1.

epsilon

Small float added to the variance to avoid dividing by 0.

center

If TRUE, add beta offset to normalized tensor.

scale

If TRUE, multiply by gamma.

betaInitializer

Intializer for the beta weight.

gammaInitializer

Intializer for the gamma weight.

betaRegularizer

Regularizer for the beta weight.

gammaRegularizer

Regularizer for the gamma weight.

betaConstraint

Optional constraint for the beta weight.

gammaConstraint

Optional constraint for the gamma weight.

Details

$initialize instantiates a new class.

$call main body.

$compute_output_shape computes the output shape.

Author(s)

Tustison NJ

Examples

InstanceNormalizationLayer$new()
InstanceNormalizationLayer$new(axis = 2L)
testthat::expect_error(InstanceNormalizationLayer$new(axis = 1L))

ANTsX/ANTsRNet documentation built on April 28, 2024, 12:16 p.m.