yongquan-qu / CBAM.Flax

Non-official implement of Paper：CBAM: Convolutional Block Attention Module using Flax and JAX

CBAM.Flax

This is a re-implement of paper CBAM: Convolutional Block Attention Module using JAX and Flax.

Woo, S., Park, J., Lee, J., & Kweon, I. (2018). CBAM: Convolutional Block Attention Module. Computer Vision – ECCV 2018, 3-19. arXiv:[https://arxiv.org/abs/1807.06521]