tfm.vision.preprocess_ops.color_jitter

Applies color jitter to an image, similarly to torchvision`s ColorJitter.

image (tf.Tensor): Of shape [height, width, 3] and type uint8. brightness (float, optional): Magnitude for brightness jitter. Defaults to 0. contrast (float, optional): Magnitude for contrast jitter. Defaults to 0. saturation (float, optional): Magnitude for saturation jitter. Defaults to 0. seed (int, optional): Random seed. Defaults to None.

tf.Tensor The augmented image of type uint8.