随机旋转层

[源代码]

RandomRotation 类

tf_keras.layers.RandomRotation(
    factor,
    fill_mode="reflect",
    interpolation="bilinear",
    seed=None,
    fill_value=0.0,
    **kwargs
)

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

此层将对每个图像应用随机旋转，并根据 fill_mode 填充空白区域。

默认情况下，随机旋转仅在训练期间应用。在推理时，该层不执行任何操作。如果您需要在推理时应用随机旋转，则在调用该层时将 training 设置为 True。

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

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

输入形状

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

输出形状

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

参数

• factor：表示 2 Pi 的分数的浮点数，或大小为 2 的元组，表示顺时针和逆时针旋转的下限和上限。正值表示逆时针旋转，负值表示顺时针旋转。当表示为单个浮点数时，此值用于上限和下限。例如，factor=(-0.2, 0.3) 会导致输出旋转随机量，范围为 [-20% * 2pi, 30% * 2pi]。factor=0.2 会导致输出旋转随机量，范围为 [-20% * 2pi, 20% * 2pi]。
• fill_mode：根据给定的模式（{"constant", "reflect", "wrap", "nearest"} 中的一个）填充输入边界之外的点。
  • reflect：(d c b a | a b c d | d c b a) 输入通过关于最后一个像素的边缘反射来扩展。
  • constant：(k k k k | a b c d | k k k k) 输入通过用相同的常数值 k = 0 填充边缘之外的所有值来扩展。
  • wrap：(a b c d | a b c d | a b c d) 输入通过环绕到相对边缘来扩展。
  • nearest：(a a a a | a b c d | d d d d) 输入通过最近的像素来扩展。
• interpolation：插值模式。支持的值："nearest"、"bilinear"。
• seed：整数。用于创建随机种子。
• fill_value：浮点数，表示当 fill_mode="constant" 时填充边界之外的值。