Keras 3 API 文档 / 层 API / 重塑层 / ZeroPadding2D 层

ZeroPadding2D 层

[源代码]

ZeroPadding2D

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

用于二维输入(例如图像)的零填充层。

此层可以在图像张量的顶部、底部、左侧和右侧添加零行和零列。

示例

>>> 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)