RandomFlip 層

[原始碼]

RandomFlip 類別

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

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

此層將根據 mode 屬性水平和/或垂直翻轉圖像。 在推論時，輸出將與輸入相同。 請使用 training=True 來呼叫此層以翻轉輸入。

輸入像素值可以是任何範圍（例如 [0., 1.) 或 [0, 255]），並且可以是整數或浮點數 dtype。 預設情況下，此層會輸出浮點數。

有關預處理層的概述和完整列表，請參閱預處理指南。

輸入形狀

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：整數。 用於建立隨機種子。