TFGENZOO.layers.spadebn module¶

class TFGENZOO.layers.spadebn.SpadeBN(width: int = 128, kernel_size=3, **kwargs)[source]¶

Bases: tensorflow.python.keras.engine.base_layer.Layer

SPADE BatchNormalization

Sources:

https://towardsdatascience.com/implementing-spade-using-fastai-6ad86b94030a

call(x: tensorflow.python.framework.ops.Tensor, cond: tensorflow.python.framework.ops.Tensor)[source]¶

This is where the layer’s logic lives.

Parameters

inputs – Input tensor, or list/tuple of input tensors.
**kwargs – Additional keyword arguments.

Returns

A tensor or list/tuple of tensors.

get_config()[source]¶

Returns the config of the layer.

A layer config is a Python dictionary (serializable) containing the configuration of a layer. The same layer can be reinstantiated later (without its trained weights) from this configuration.

The config of a layer does not include connectivity information, nor the layer class name. These are handled by Network (one layer of abstraction above).

Returns: Python dictionary.