Reading: CBAM — Convolutional Block Attention Module (Image Classification)

Image for post
Image for post

Outline

1. CBAM: General Architecture

Image for post
Image for post
CBAM: General Architecture
Image for post
Image for post
Image for post
Image for post

2. Channel Attention Module

Image for post
Image for post
Channel Attention Module
Image for post
Image for post

3. Spatial Attention Module

Image for post
Image for post
Spatial Attention Module
Image for post
Image for post

4. Ablation Study on ImageNet

4.1. Max Pool or Avg Pool

Image for post
Image for post
Comparison of different channel attention methods

4.2. Spatial and Channel Attention

Image for post
Image for post
Comparison of different spatial attention methods

4.3. Arrangement of the Channel and Spatial Attention

Image for post
Image for post

5. SOTA Comparison

5.1. ImageNet

Image for post
Image for post
Classification results on ImageNet-1K.
Image for post
Image for post
Classification results on ImageNet-1K using the light-weight network, MobileNet

5.2. Network Visualization with Grad-CAM

Image for post
Image for post
Grad-CAM visualization results

5.4. MS COCO Object Detection

Image for post
Image for post
Object detection mAP(%) on the MS COCO validation set

5.5. VOC 2007 Object Detection

Image for post
Image for post
PASCAL VOC 2007 test set

PhD, Researcher. I share what I've learnt and done. :) My LinkedIn: https://www.linkedin.com/in/sh-tsang/, My Paper Reading List: https://bit.ly/33TDhxG

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store