python - Tensorflow:如何池化深度？

Question

我定义了以下参数，用于在图像深度 (rgb) 上进行最大池化，以便在密集层和读出之前进行压缩……我失败了，错误是我无法在深度和其他所有方面进行池化:

sunset_poolmax_1x1x3_div_2x2x3_params = \
    {'pool_function':tf.nn.max_pool,
     'ksize':[1,1,1,3],
     'strides':[1,1,1,3],
     'padding': 'SAME'}

我将步幅更改为 [1,1,1,3]，因此深度是池减少的唯一维度……但它仍然不起作用。我无法通过必须压缩所有内容以保持颜色的微小图像获得良好的结果...

实际错误:

ValueError: Current implementation does not support pooling in the batch and depth dimensions.

Answer 1

tf.nn.max_pool 不支持在深度维度上进行池化，这就是您收到错误的原因。

您可以使用最大缩减来实现您想要的:

tf.reduce_max(input_tensor, reduction_indices=[3], keep_dims=True)

上面的 keep_dims 参数确保张量的等级得以保留。这确保最大缩减的行为与 tf.nn.max_pool 操作在支持深度维度上的池化时将执行的操作一致。