RandomFlip 層

[原始碼]

RandomFlip 類別

keras.layers.RandomFlip(
    mode="horizontal_and_vertical", seed=None, data_format=None, **kwargs
)

一個預處理層，可在訓練期間隨機翻轉圖像。

此層將根據 mode 屬性水平和/或垂直翻轉圖像。在推論期間，輸出將與輸入相同。使用 training=True 呼叫此層以翻轉輸入。輸入像素值可以是任何範圍（例如 [0., 1.) 或 [0, 255]），並且可以是整數或浮點數 dtype。預設情況下，此層將輸出浮點數。

注意： 此層可以安全地在 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。