View source on GitHub |
Implements Autocontrast function from PIL using TF ops.
tfm.vision.augment.autocontrast(
image: tf.Tensor
) -> tf.Tensor
Args | |
---|---|
image
|
A 3D uint8 tensor. |
Returns | |
---|---|
The image after it has had autocontrast applied to it and will be of type uint8. |