Мне нужен (не обучаемый) слой в Керасе, который выполняет локальный максимум при поиске 2-мерного входа и суммирует все записи в определенной среде вокруг этого максимума. Форма ввода для слоя выглядит примерно так (Нет, 1, 24, 16).
Чтобы было понятнее, давайте посмотрим, как будет выглядеть ввод:
Тогда желаемый результат слоя должен быть:
Таким образом, он должен найти каждый локальный максимум и затем суммировать все записи от (до) 8 соседних пикселей до этого максимума.
Это проблема, связанная с физикой высоких энергий, поэтому я думаю, что интерпретация этих «образов» не должна быть слишком важной.
Как мне добиться этого, используя слои Keras?
Есть предложения? Спасибо!