张量流中的自定义中值池

Question

我正在尝试在张量流中实现中值池化层。

然而，两者tf.nn.median_pool都不存在tf.reduce_median。

有没有办法用python api来实现这种池化层？

Answer 1

您可以使用类似：

patches = tf.extract_image_patches(tensor, [1, k, k, 1], ...)
m_idx = int(k*k/2+1)
top = tf.top_k(patches, m_idx, sorted=True)
median = tf.slice(top, [0, 0, 0, m_idx-1], [-1, -1, -1, 1])

为了容纳均匀大小的中值内核和多个通道，您将需要扩展它，但这应该可以为您提供大部分帮助。