RandomFlip 层

[源]

RandomFlip

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

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

该层将根据 mode 属性水平翻转或垂直翻转图像。在推理时,输出将与输入相同。调用该层时将 training 设置为 True 即可翻转输入。

输入像素值可以是任何范围(例如 [0., 1.)[0, 255]),且可以是整数或浮点类型。默认情况下,该层将输出浮点数。

有关预处理层的概述和完整列表,请参阅预处理指南

输入形状

形状为 (..., height, width, channels) 的 3D(非批处理)或 4D(批处理)张量,格式为 "channels_last"

输出形状

形状为 (..., height, width, channels) 的 3D(非批处理)或 4D(批处理)张量,格式为 "channels_last"

参数

  • mode: 字符串,指示要使用的翻转模式。可以是 "horizontal""vertical""horizontal_and_vertical""horizontal" 是左右翻转,"vertical" 是上下翻转。默认为 "horizontal_and_vertical"
  • seed: 整数。用于创建随机种子。