Rescaling
classkeras.layers.Rescaling(scale, offset=0.0, **kwargs)
一个预处理层,用于将输入值重新缩放到新的范围.
该层通过乘以 scale
并加上 offset
来重新缩放输入的每个值(通常是图像).
例如:
要将输入范围在 [0, 255]
内的值重新缩放到 [0, 1]
范围内,你可以传递 scale=1./255
.
要将输入范围在 [0, 255]
内的值重新缩放到 [-1, 1]
范围内,你可以传递 scale=1./127.5, offset=-1
.
重新缩放在训练和推理期间都会应用.输入可以是整数或浮点数类型,默认情况下,该层将输出浮点数.
注意: 该层在 tf.data
管道中使用是安全的(与您使用的后端无关).
参数:
scale: 浮点数,应用于输入的比例.
offset: 浮点数,应用于输入的偏移量.
**kwargs: 基础层的键值对参数,例如 name
和 dtype
.