當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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


用於 2D 輸入的上采樣層。

繼承自:LayerModule

用法

tf.keras.layers.UpSampling2D(
    size=(2, 2), data_format=None, interpolation='nearest', **kwargs
)

參數

  • size Int,或 2 個整數的元組。行和列的上采樣因子。
  • 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"。
  • interpolation 一個字符串,是 nearestbilinear 之一。

分別通過size[0]size[1]重複數據的行和列。

例子:

input_shape = (2, 2, 1, 3)
x = np.arange(np.prod(input_shape)).reshape(input_shape)
print(x)
[[[[ 0  1  2]]
  [[ 3  4  5]]]
 [[[ 6  7  8]]
  [[ 9 10 11]]]]
y = tf.keras.layers.UpSampling2D(size=(1, 2))(x)
print(y)
tf.Tensor(
  [[[[ 0  1  2]
     [ 0  1  2]]
    [[ 3  4  5]
     [ 3  4  5]]]
   [[[ 6  7  8]
     [ 6  7  8]]
    [[ 9 10 11]
     [ 9 10 11]]]], shape=(2, 2, 2, 3), 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, upsampled_rows, upsampled_cols, channels)
  • 如果 data_format"channels_first"(batch_size, channels, upsampled_rows, upsampled_cols)

相關用法


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