Rescaling 层

[源代码]

Rescaling

keras.layers.Rescaling(scale, offset=0.0, **kwargs)

一个将输入值重缩放到新范围的预处理层。

此层通过乘以 `scale` 并加上 `offset` 来重缩放输入的每个值(通常是图像)。

例如

  1. 要将 `[0, 255]` 范围内的输入重缩放到 `[0, 1]` 范围,您可以传递 `scale=1./255`。

  2. 要将 `[0, 255]` 范围内的输入重缩放到 `[-1, 1]` 范围,您可以传递 `scale=1./127.5, offset=-1`。

重缩放应用于训练和推断。输入可以是整数或浮点 dtype,默认情况下,该层将输出浮点数。

注意:此层可以在 tf.data 管道中使用(与您使用的后端无关)。

参数

  • scale:浮点数、整数、列表、元组或 np.ndarray。应用于输入的缩放因子。如果为标量,则相同的缩放因子将应用于输入的所有特征或通道。如果为列表、元组或一维数组,则按通道应用缩放。
  • offset:浮点数、整数、列表/元组或 numpy ndarray。应用于输入的偏移量。如果为标量,则相同的偏移量将应用于输入的所有特征或通道。如果为列表、元组或一维数组,则按通道应用偏移。
  • **kwargs:基础层关键字参数,例如 namedtype