CBAM.PyTorch
论文:CBAM的非官方实现者:卷积块注意模块
介绍
这些代码是用于纸张的重新实现版本:CBAM:卷积块注意模块
Woo S,Park J,Lee JY等。 CBAM:卷积块注意模块
结构
CBAM概述。 该模块具有两个连续的子模块:通道和空间。 中间特征图通过我们的模块(CBAM)在深度网络的每个卷积块上进行自适应调整。
要求
Python3
PyTorch 0.4.1
tensorboardX(可选)
火炬网
预训练模型(可选)
结果
我们仅在ImageNet-1K中测试了四个模型,训练集和验证集都缩放到256(最小边),仅使用Mirror和RandomResizeCrop作为训练数据扩充,在验证过程中,我们使用中心裁剪获得224x224补丁。
ImageNet-1K
楷模
验证(Top-1)
验证(前5名)
ResNet50
74.26
91
2022-03-09 20:03:23
66KB
Python
1