Search code examples
imagekerasmodellayerrescale

Add a rescaling layer to a keras model


Suppose I have an input image of shape (100, 100, 3) and I want to rescale the image, i.e., make it smaller and have a dimension say (50, 50, 3), i.e., rescaling it to half. Would be better if Gaussian smoothing is applied to the image within the layer after rescaling.


Solution

  • Ahh, this works fine, I figured it out,

    y = layers.Lambda( 
            lambda image: tf.image.resize( 
                inputs, 
                (int(80 * 1/2), int(80 * 1/2)), 
                method = tf.image.ResizeMethod.BICUBIC,
                #align_corners = True, # possibly important
                preserve_aspect_ratio = True
            )
        )(inputs)