How to implement SegNet with preserving max-indexes in Keras

问题

I'm trying to implement SegNet in Keras (tf backend) to do semantic segmentation.

The most impressived trick of SgeNet is to pass max-pooling indices to the upsampling layers. However, there are many implementations of SegNet in Keras(e.g.) I find on github just using simple UpSampling (called SegNet-Basic).

I notice that it can be achieved in Tensorflow with " tf.nn.max_pool_with_argmax ". So I want to know is there any similar method to get the max-pooling indices and put them back in upsampling in Keras.

Thanks in advance.

回答1:

Well, I think I've found the answer.

来源：https://stackoverflow.com/questions/50924072/how-to-implement-segnet-with-preserving-max-indexes-in-keras