asdf2kr / BAM-CBAM-pytorch

Pytorch implementation of BAM("BAM: Bottleneck Attention Module", BMVC18) and CBAM(“CBAM: Convolutional Block Attention Module”, ECCV18)
MIT License
31 stars 4 forks source link