Keras 3 API 文件 / 層 API / 形狀重塑層 / ZeroPadding2D 層

ZeroPadding2D 層

[原始碼]

ZeroPadding2D 類別

keras.layers.ZeroPadding2D(padding=(1, 1), data_format=None, **kwargs)

用於 2D 輸入(例如圖片)的零填充層。

此層可以在圖像張量的頂部、底部、左側和右側添加零行和零列。

範例

>>> input_shape = (1, 1, 2, 2)
>>> x = np.arange(np.prod(input_shape)).reshape(input_shape)
>>> x
[[[[0 1]
   [2 3]]]]
>>> y = keras.layers.ZeroPadding2D(padding=1)(x)
>>> y
[[[[0 0]
   [0 0]
   [0 0]
   [0 0]]
  [[0 0]
   [0 1]
   [2 3]
   [0 0]]
  [[0 0]
   [0 0]
   [0 0]
   [0 0]]]]

參數

  • padding:整數,或 2 個整數的元組,或 2 個 2 個整數的元組。
    • 若為整數:則對高度和寬度應用相同的對稱填充。
    • 若為 2 個整數的元組:則解釋為高度和寬度的兩個不同的對稱填充值:(symmetric_height_pad, symmetric_width_pad)
    • 若為 2 個 2 個整數的元組:則解釋為 ((top_pad, bottom_pad), (left_pad, right_pad))
  • data_format:字串,為 "channels_last"(預設)或 "channels_first" 其中之一。輸入中維度的順序。"channels_last" 對應於形狀為 (batch_size, height, width, channels) 的輸入,而 "channels_first" 對應於形狀為 (batch_size, channels, height, width) 的輸入。未指定時,使用在您的 Keras 設定檔 ~/.keras/keras.json(如果存在)中找到的 image_data_format 值。預設為 "channels_last"

輸入形狀

4D 張量,形狀為:- 若 data_format"channels_last"(batch_size, height, width, channels) - 若 data_format"channels_first"(batch_size, channels, height, width)

輸出形狀

4D 張量,形狀為:- 若 data_format"channels_last"(batch_size, padded_height, padded_width, channels) - 若 data_format"channels_first"(batch_size, channels, padded_height, padded_width)