RandomCrop 層

[原始碼]

RandomCrop 類別

tf_keras.layers.RandomCrop(height, width, seed=None, **kwargs)

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

在訓練期間,此層會隨機選擇一個位置,將影像裁剪至目標大小。此層會將同一批次中的所有影像裁剪到相同的裁剪位置。

在推論時,以及在訓練期間,如果輸入影像小於目標大小,則會調整輸入大小並裁剪,以便返回影像中與目標長寬比匹配的最大可能視窗。 如果您需要在推論時應用隨機裁剪,請在呼叫此層時將 training 設定為 True。

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

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

輸入形狀

具有形狀的 3D(未批次處理)或 4D(批次處理)張量:(..., height, width, channels),採用 "channels_last" 格式。

輸出形狀

具有形狀的 3D(未批次處理)或 4D(批次處理)張量:(..., target_height, target_width, channels)

參數

  • height:整數,輸出形狀的高度。
  • width:整數,輸出形狀的寬度。
  • seed:整數。用於建立隨機種子。