HotoNagi / Chainer-CBAM

Chainer-CBAM(Convolutional Block Attention Module)

Training

python n_fold_train.py --dataset [DATASET] --arch [MODEL] --gpu [GPU_ID]

軽い解説

論文リンク

使用ネットワーク

ResBlockに以下の図のような Channel Attention Module , Spatial Attention Module と呼ばれる2つの Attention Module を追加したもの。

Channel Attention Module

Spatial Attention Module

参考実装

• 公式（pytorch）
• 非公式（pytorch）