Keras 2 API 文件 / 層 API / 預處理層 / 影像增強層 / RandomRotation 層

RandomRotation 層

[原始碼]

RandomRotation 類別

tf_keras.layers.RandomRotation(
    factor,
    fill_mode="reflect",
    interpolation="bilinear",
    seed=None,
    fill_value=0.0,
    **kwargs
)

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

此層將對每個影像應用隨機旋轉,並根據 fill_mode 填充空白區域。

預設情況下,隨機旋轉僅在訓練期間應用。在推論時,此層不會執行任何操作。如果您需要在推論時應用隨機旋轉,請在呼叫此層時將 training 設定為 True。

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

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

輸入形狀

3D (未批次) 或 4D (已批次) 張量,形狀為:(..., height, width, channels),採用 "channels_last" 格式

輸出形狀

3D (未批次) 或 4D (已批次) 張量,形狀為:(..., height, width, channels),採用 "channels_last" 格式

引數

  • factor:一個浮點數,表示 2 Pi 的比例,或大小為 2 的元組,表示順時針和逆時針旋轉的下限和上限。正值表示逆時針旋轉,而負值表示順時針旋轉。當表示為單個浮點數時,此值將用於上限和下限。例如,factor=(-0.2, 0.3) 會導致輸出旋轉一個介於 [-20% * 2pi, 30% * 2pi] 範圍內的隨機量。factor=0.2 會導致輸出旋轉一個介於 [-20% * 2pi, 20% * 2pi] 範圍內的隨機量。
  • fill_mode:根據給定的模式({"constant", "reflect", "wrap", "nearest"} 之一),填充輸入邊界外的點。
    • reflect(d c b a | a b c d | d c b a) 輸入透過反射最後一個像素的邊緣來延伸。
    • constant(k k k k | a b c d | k k k k) 輸入透過使用相同的常數值 k = 0 填充邊緣之外的所有值來延伸。
    • wrap(a b c d | a b c d | a b c d) 輸入透過環繞到相對的邊緣來延伸。
    • nearest(a a a a | a b c d | d d d d) 輸入透過使用最近的像素來延伸。
  • interpolation:插值模式。支援的值:"nearest""bilinear"
  • seed:整數。用於建立隨機種子。
  • fill_value:一個浮點數,表示當 fill_mode="constant" 時,要填充到邊界之外的值。