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.
Input pixel values can be of any range (e.g. [0., 1.)
or [0, 255]
) and
of integer or floating point dtype.
By default, the layer will output floats.
For an overview and full list of preprocessing layers, see the preprocessing guide.
Input shape | |
---|---|
3D
|
unbatched) or 4D (batched) tensor with shape
|
Output shape | |
---|---|
3D
|
unbatched) or 4D (batched) tensor with shape
|