RandomFlip 层

[源代码]

RandomFlip 类

keras.layers.RandomFlip(
    mode="horizontal_and_vertical", seed=None, data_format=None, **kwargs
)

一个在训练期间随机翻转图像的预处理层。

此层将根据 mode 属性水平和/或垂直翻转图像。在推理时，输出将与输入相同。使用 training=True 调用此层以翻转输入。输入像素值可以是任何范围（例如 [0., 1.) 或 [0, 255]），并且可以是整数或浮点 dtype。默认情况下，该层将输出浮点数。

注意： 此层可以安全地在 tf.data 管道内部使用（独立于您使用的后端）。

输入形状

3D（非批次）或 4D（批次）张量，形状为：(..., height, width, channels)，采用 "channels_last" 格式。

输出形状

3D（非批次）或 4D（批次）张量，形状为：(..., height, width, channels)，采用 "channels_last" 格式。

参数

• mode: 字符串，指示要使用的翻转模式。可以是 "horizontal"、"vertical" 或 "horizontal_and_vertical"。"horizontal" 是左右翻转，"vertical" 是上下翻转。默认为 "horizontal_and_vertical"
• seed: 整数。用于创建随机种子。
• **kwargs: 基础层关键字参数，例如 name 和 dtype。