image-processing - Keras 函数的 Tensorflow 等效项：UpSampling2D

Question

我想使用 Keras 层：

from keras.layers.convolutional import UpSampling2D
x = UpSampling2D((2, 2))(x)

如何使用本机 tensorflow 复制此行为？

我无法找到等效的功能/层。

Answer 1

假设x是 shape (BATCH_SIZE, H, W, C)，你可以使用tf.image.resize_nearest_neighbor，这是 keras 使用的后端实现：

x = tf.image.resize_nearest_neighbor(x, (2*H,2*W))

Answer 2

有 tf.keras.layers.UpSampling2D。我不太确定，但我认为 tf.image 函数仅在 CPU 上实现。