ResNetBlock

Inheritance Diagram

Inheritance diagram of ashpy.models.convolutional.pix2pixhd.ResNetBlock

class ashpy.models.convolutional.pix2pixhd.ResNetBlock(filters, normalization_layer=<class 'ashpy.layers.instance_normalization.InstanceNormalization'>, non_linearity=<class 'tensorflow.python.keras.layers.advanced_activations.ReLU'>, kernel_size=3, num_blocks=2)[source]

Bases: tensorflow.python.keras.engine.training.Model

ResNet Blocks.

The input filters is the same as the output filters.

Methods

__init__(filters[, normalization_layer, …]) Build the ResNet block composed by num_blocks.
call(inputs[, training]) Forward pass.

Attributes

activity_regularizer Optional regularizer function for the output of this layer.
dtype
dynamic
inbound_nodes Deprecated, do NOT use! Only for compatibility with external Keras.
input Retrieves the input tensor(s) of a layer.
input_mask Retrieves the input mask tensor(s) of a layer.
input_shape Retrieves the input shape(s) of a layer.
input_spec Gets the network’s input specs.
layers
losses Losses which are associated with this Layer.
metrics Returns the model’s metrics added using compile, add_metric APIs.
metrics_names Returns the model’s display labels for all outputs.
name Returns the name of this module as passed or determined in the ctor.
name_scope Returns a tf.name_scope instance for this class.
non_trainable_variables
non_trainable_weights
outbound_nodes Deprecated, do NOT use! Only for compatibility with external Keras.
output Retrieves the output tensor(s) of a layer.
output_mask Retrieves the output mask tensor(s) of a layer.
output_shape Retrieves the output shape(s) of a layer.
run_eagerly Settable attribute indicating whether the model should run eagerly.
sample_weights
state_updates Returns the updates from all layers that are stateful.
stateful
submodules Sequence of all sub-modules.
trainable
trainable_variables Sequence of variables owned by this module and it’s submodules.
trainable_weights
updates
variables Returns the list of all layer variables/weights.
weights Returns the list of all layer variables/weights.
__init__(filters, normalization_layer=<class 'ashpy.layers.instance_normalization.InstanceNormalization'>, non_linearity=<class 'tensorflow.python.keras.layers.advanced_activations.ReLU'>, kernel_size=3, num_blocks=2)[source]

Build the ResNet block composed by num_blocks.

Each block is composed by
  • Conv2D with strides 1 and padding “same”
  • Normalization Layer
  • Non Linearity

The final result is the output of the ResNet + input.

Parameters:
  • filters (int) – initial filters (same as the output filters).
  • normalization_layer (tf.keras.layers.Layer) – layer of normalization used by the residual block.
  • non_linearity (tf.keras.layers.Layer) – non linearity used in the resnet block.
  • kernel_size (int) – kernel size used in the resnet block.
  • num_blocks (int) – number of blocks, each block is composed by conv, normalization and non linearity.
call(inputs, training=False)[source]

Forward pass.

Parameters:
  • inputs – input tensor.
  • training – whether is training or not.
Returns:

A Tensor of the same shape as the inputs. The input passed through num_blocks blocks.