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

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

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

参数

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