def image_augmentation(img,label):
  img,label = normalize_img(img,label)
  img = img_flip_vertical(img)
  img = img_random_contrast(img)
  img = img_random_brightness(img)
  img = random_hue(img)
  return img,label

def normalize_img(img,label):
  img = tf.cast(img,tf.float32)/255.0
  return img,label

        __
        __