def image_augmentation(img,label):
  img = tf.image.flip_left_right(img)
  img = tf.image.random_brightness(img, max_delta=0.2)
  img =tf.image.random_contrast(img,lower=0.8, upper=1.2)
  img = normalize_img(img,label)
  return img,label

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

        __
        __