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
|
View source on GitHub