Skip to content

Non-official implement of Paper:CBAM: Convolutional Block Attention Module

Notifications You must be signed in to change notification settings

luuuyi/CBAM.PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CBAM.PyTorch

Non-official implement of Paper:CBAM: Convolutional Block Attention Module

Introduction

The codes are PyTorch re-implement version for paper: CBAM: Convolutional Block Attention Module

Woo S, Park J, Lee J Y, et al. CBAM: Convolutional Block Attention Module[J]. 2018. ECCV2018

Structure

The overview of CBAM. The module has two sequential sub-modules: channel and spatial. The intermediate feature map is adaptively refined through our module (CBAM) at every convolutional block of deep networks.

1

Requirements

  • Python3
  • PyTorch 0.4.1
  • tensorboardX (optional)
  • torchnet
  • pretrainedmodels (optional)

Results

We just test four models in ImageNet-1K, both train set and val set are scaled to 256(minimal side), only use Mirror and RandomResizeCrop as training data augmentation, during validation, we use center crop to get 224x224 patch.

ImageNet-1K

Models validation(Top-1) validation(Top-5)
ResNet50 74.26 91.91
ResNet50-CBAM 75.45 92.55

About

Non-official implement of Paper:CBAM: Convolutional Block Attention Module

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published