/BAM-CBAM-pytorch

Pytorch implementation of BAM("BAM: Bottleneck Attention Module", BMVC18) and CBAM(“CBAM: Convolutional Block Attention Module”, ECCV18)

Primary LanguagePythonMIT LicenseMIT

Stargazers