Python tf.keras.layers.ZeroPadding2D用法及代碼示例

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

繼承自：Layer，Module

用法

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

參數

padding
整數，或 2 個整數的元組，或 2 個整數的 2 個元組。
- 如果 int: 相同的對稱填充應用於高度和寬度。
- 如果 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 配置文件中的 image_data_format 值 ~/.keras/keras.json 。如果您從未設置它，那麽它將是"channels_last"。

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

例子：

input_shape = (1, 1, 2, 2)
x = np.arange(np.prod(input_shape)).reshape(input_shape)
print(x)
[[[[0 1]
   [2 3]]]]
y = tf.keras.layers.ZeroPadding2D(padding=1)(x)
print(y)
tf.Tensor(
  [[[[0 0]
     [0 0]
     [0 0]
     [0 0]]
    [[0 0]
     [0 1]
     [2 3]
     [0 0]]
    [[0 0]
     [0 0]
     [0 0]
     [0 0]]]], shape=(1, 3, 4, 2), dtype=int64)

輸入形狀：

具有形狀的 4D 張量：

如果 data_format 是 "channels_last" ： (batch_size, rows, cols, channels)
如果 data_format 是 "channels_first" ： (batch_size, channels, rows, cols)

輸出形狀：

具有形狀的 4D 張量：

如果 data_format 是 "channels_last" ： (batch_size, padded_rows, padded_cols, channels)
如果 data_format 是 "channels_first" ： (batch_size, channels, padded_rows, padded_cols)

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.layers.ZeroPadding2D。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。