tensorflow - TensorFlow 中的上采样特征图

标签 tensorflow conv-neural-network

我想为图像分割项目实现一个卷积-反卷积网络。在反卷积部分,我计划将特征图上采样 2。例如原始特征图的维度为 64*64*4,我想将其上采样为 128*128*4。有谁知道这样做的张量操作?谢谢!

最佳答案

您可以使用 tf.image.resize_images() .它需要批量图像或单张图像,并支持最常见的方法,例如bilinearnearest_neighbor .

这是 TensorFlow API 引用的链接:resizing

也可以看看tflearn等更高级别的API中是如何实现上采样操作的.您可以找到 upsample_2dupscore_layer在他们的 Github 仓库中:conv.py

注:输出可能会强制转换为 tf.float32在较旧的 TF 版本中

关于tensorflow - TensorFlow 中的上采样特征图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36547615/

相关文章:

python - 如何使用 tf.while_loop 进行急切执行?

python - 如何将 TensorFlow 2D 卷积 (tf.nn.conv2d) 应用于单个(非批量)2D 图像?

python - 使用经过训练的 ConvNet 的参数预测图像类别

machine-learning - 如何制作多尺度图像来训练 CNN

python - pytorch 中的 Keras 学习率衰减

python - 为CNN编写训练模型

tensorflow - 在温度预测方面哪种损失函数比 MSE 更好?

c++ - 如何使用 Custom OP 在 C++ 中构建 TensorFlow Graph?

python - 无法从 tensorflow /keras 中加载的模型获取梯度

python - Tensorflow Estimator API 在评估模式下保存图像摘要