Source code for pytwovision.models.layers.upsample_layer

import tensorflow as tf

[docs]class UpsampleLayer(tf.keras.layers.Layer):
    """ A layer for upsampling an image. """
[docs]    def call(self, inputs):
        return tf.image.resize(inputs, (inputs.shape[1] * 2, inputs.shape[2] * 2), method='nearest')