RandomFlip 层

[源代码]

RandomFlip

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

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

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

注意:该层可以在 tf.datagrain 管道中使用(无论您使用的是哪个后端),且是安全的。

输入形状

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