View source on GitHub |
Jitters saturation of an image, similarly to torchvision`s ColorJitter.
tfm.vision.preprocess_ops.random_saturation(
image: tf.Tensor, saturation: float = 0.0, seed: Optional[int] = None
) -> tf.Tensor
Args | |
---|---|
image
|
tf.Tensor
Of shape [height, width, 3] and type uint8. |
saturation
|
float, optional
Magnitude for saturation jitter. Defaults to 0. |
seed
|
int, optional
Random seed. Defaults to None. |
Returns | |
---|---|
tf.Tensor
|
The augmented image of type uint8.
|