View source on GitHub |
A preprocessing layer which randomly flips images during training.
tf.keras.layers.RandomFlip(
mode=HORIZONTAL_AND_VERTICAL, seed=None, **kwargs
)
This layer will flip the images horizontally and or vertically based on the
mode
attribute. During inference time, the output will be identical to
input. Call the layer with training=True
to flip the input.
For an overview and full list of preprocessing layers, see the preprocessing guide.
Input shape | |
---|---|
3D (unbatched) or 4D (batched) tensor with shape:
(..., height, width, channels) , in "channels_last" format.
|
Output shape | |
---|---|
3D (unbatched) or 4D (batched) tensor with shape:
(..., height, width, channels) , in "channels_last" format.
|