重缩放层

[源代码]

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

重缩放在训练和推理期间均适用。输入可以是整数或浮点类型,默认情况下,该层将输出浮点数。

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

参数

  • scale:浮点数,应用于输入的缩放比例。
  • offset:浮点数,应用于输入的偏移量。
  • **kwargs**:基础层关键字参数,例如namedtype