我正在尝试在 Keras(tf 后端)中实现 SegNet 来进行语义分割。
SgeNet 最令人印象深刻的技巧是将最大池索引传递给上采样层。但是,我在 github 上找到了许多 Keras 中的 SegNet 实现(例如),仅使用简单的 UpSampling(称为 SegNet-Basic)。
我注意到它可以在 Tensorflow 中使用“ tf.nn.max_pool_with_argmax ”来实现。所以我想知道是否有任何类似的方法来获取最大池索引并将它们放回 Keras 中的上采样中。
提前致谢。