如何在 Keras 中通过保留最大索引来实现 SegNet

Question

我正在尝试在 Keras（tf 后端）中实现 SegNet 来进行语义分割。

SgeNet 最令人印象深刻的技巧是将最大池化索引传递给上采样层。但是，我在 github 上发现 Keras(e.g.) 中有很多 SegNet 的实现，只是使用简单的上采样（称为 SegNet-Basic）。

我注意到它可以在 Tensorflow 中使用“tf.nn.max_pool_with_argmax”来实现。所以我想知道是否有任何类似的方法来获取最大池索引并将它们放回 Keras 的上采样中。

提前致谢。

Answer 1

嗯，我想我找到了 answer。