tensorflow 实现 unpool(反池化)
网上很多人都说TensorFlow中目前还没有反池化操作的函数,实际上是有的,只是命名不是 tf.nn.max_unpool,而是 gen_nn_ops.max_pool_grad,需从tensorflow.python.ops中导入。从接口的命名可以看出,max_pool_grad是在计算最大池化反向传播时的梯度,事实上max unpool就是max pool的梯度。示例代码如下:
import tensorflow as tf
from tensorflow.python.ops import gen_nn_ops
#最大池化
pool1 = tf.nn.max_pool(input,
ksize=[1, 2, 2,