input_layer = tf.keras.Input(shape=(img_width, img_height, img_channels), name="input_layer")
conv1 = tf.keras.layers.Conv2D(filters=32,
                               kernel_size= 2,
                               strides = 2,
                               padding="same",
                               data_format="channels_last",
                               activation=tf.nn.relu)(input_layer)