Как реализовать SegNet с сохранением max-индексов в Keras

Я пытаюсь реализовать SegNet в Keras (tf backend) для семантической сегментации.

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

Я заметил, что это может быть достигнуто в Tensorflow с помощью "tf.nn.max_pool_with_argmax". Итак, я хочу знать, есть ли какой-либо аналогичный метод для получения индексов максимального объединения и возврата их в повышающую дискретизацию в Керасе.

Заранее спасибо.


person Ryan Ye    schedule 19.06.2018    source источник


Ответы (1)


Что ж, я думаю, что нашел ответ.

person Ryan Ye    schedule 20.06.2018