gpt4 book ai didi

image-processing - Tensorflow 等效于 Keras 函数 : UpSampling2D

转载 作者:行者123 更新时间:2023-12-04 13:56:59 26 4
gpt4 key购买 nike

我想使用 Keras 层:

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

如何使用 native tensorflow 复制此行为?

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

最佳答案

假设 x形状(BATCH_SIZE, H, W, C) ,您可以使用 tf.image.resize_nearest_neighbor ,这是keras使用的后端实现:

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

关于image-processing - Tensorflow 等效于 Keras 函数 : UpSampling2D,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46677105/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com