重缩放层

[source]

Rescaling class

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.