Bagaimana menerapkan SegNet dengan mempertahankan indeks maksimal di Keras

Saya mencoba mengimplementasikan SegNet di Keras (tf backend) untuk melakukan segmentasi semantik.

Trik SgeNet yang paling mengesankan adalah meneruskan indeks pengumpulan maksimal ke lapisan upsampling. Namun, ada banyak implementasi SegNet di Keras(mis.) Saya temukan di github hanya menggunakan UpSampling sederhana (disebut SegNet-Basic).

Saya perhatikan bahwa ini dapat dicapai di Tensorflow dengan " tf.nn.max_pool_with_argmax ". Jadi saya ingin tahu apakah ada metode serupa untuk mendapatkan indeks max-pooling dan mengembalikannya ke upsampling di Keras.

Terima kasih sebelumnya.


person Ryan Ye    schedule 19.06.2018    source sumber


Jawaban (1)


Sepertinya saya sudah menemukan jawabannya.

person Ryan Ye    schedule 20.06.2018