Я пытаюсь реализовать SegNet в Keras (tf backend) для семантической сегментации.
Самый впечатляющий трюк SgeNet - это передача индексов максимального объединения на уровни повышающей дискретизации. Однако существует множество реализаций SegNet в Keras (например) Я нахожу на github просто с помощью простого UpSampling (называемого SegNet-Basic).
Я заметил, что это может быть достигнуто в Tensorflow с помощью "tf.nn.max_pool_with_argmax". Итак, я хочу знать, есть ли какой-либо аналогичный метод для получения индексов максимального объединения и возврата их в повышающую дискретизацию в Керасе.
Заранее спасибо.