RandomFlip
classkeras.layers.RandomFlip(mode="horizontal_and_vertical", seed=None, **kwargs)
一个在训练期间随机翻转图像的预处理层.
该层将根据mode
属性水平和/或垂直翻转图像.在推理时间,输出将与输入相同.调用该层时使用training=True
来翻转输入.输入像素值可以是任何范围(例如[0., 1.)
或[0, 255]
),并且可以是整数或浮点数据类型.默认情况下,该层将输出浮点数.
注意: 该层在tf.data
管道中使用是安全的(无论您使用哪个后端).
输入形状:
具有形状的3D(未批处理)或4D(批处理)张量:
(..., height, width, channels)
,在"channels_last"
格式中.
输出形状:
具有形状的3D(未批处理)或4D(批处理)张量:
(..., height, width, channels)
,在"channels_last"
格式中.
参数:
mode: 指示要使用的翻转模式的字符串.可以是"horizontal"
、"vertical"
或"horizontal_and_vertical"
."horizontal"
是左右翻转,"vertical"
是上下翻转.默认为"horizontal_and_vertical"
seed: 整数.用于创建随机种子.
**kwargs: 基础层的键值对参数,例如name
和dtype
.