RandomFlip 层

[源代码]

RandomFlip

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

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

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

注意:此层可以在 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:基础层关键字参数,例如 namedtype