gradient_3_3

def format_image(data):
    """
       Function to reshape, format and normalize input images to make it compatible with the deep learning model
       Inputs:
       data: Input image
       Outputs:
       image: Formatted image
       data["label"]: Training label associated with the image
    """
    image = data["image"]
    image = tf.reshape(image, [-1])
    image = tf.cast(image, 'float32')
    image = image / 255.0
    return image, data["label"]