Function adaptive_instance_normalization

tflib/deep_learning/layers.py:232–240

(x, gamma=None, beta=None, epsilon=1e-5)

def adaptive_instance_normalization(x, gamma=None, beta=None, epsilon=1e-5):
    # modified from https://github.com/taki0112/MUNIT-Tensorflow/blob/master/ops.py
    # x: (N, H, W, C), gamma: (N, C), beta: (N, C)

    c_mean, c_var = tf.nn.moments(x, axes=[1, 2], keep_dims=True)
    c_std = tf.sqrt(c_var + epsilon)
    x = (x - c_mean) / c_std

    return adaptive_scaling(x, gamma, beta)


# ==============================================================================