# ResNet_Attention(CBAM,SE)
Official instruction: [CBAM](https://arxiv.org/abs/1807.06521),[SE](https://arxiv.org/abs/1709.01507)
# Required Environment
Ubuntu20.04
GTX 1080Ti
Python3.7
PyTorch1.7.0
CUDA10.2
CuDNN7.0
# Usage Method(trian with CIFAR10)
The model's backbone is ResNet. In our training, we use CIFAR10 as our dataset.
```
# To train with Se
python train_CIFAR10.py --prefix 4 --device 1 --epoch 160 --att_type se
# To trian with CBAM
python train_CIFAR10.py --prefix 5 --device 1 --epoch 160 --att_type cbam
```
# Validation Result
* ResNet50 (trained for 160 epochs) ACC@1=93.41% ACC@5=99.84%
* ResNet50+SE (trained for 160 epochs) ACC@1=94.01% ACC@5=99.89%
# Result Graph
**Blue:ResNet50**
**Red:ResNet50+SE**


没有合适的资源?快使用搜索试试~ 我知道了~
ResNet_Attention:ResNet +注意力

共16个文件
xml:8个
py:5个
iml:1个


温馨提示
ResNet_Attention(CBAM,SE) 官方说明: , 所需环境 Ubuntu20.04 GTX 1080Ti Python3.7 PyTorch 1.7.0 CUDA10.2 CuDNN7.0 使用方法(带有CIFAR10的trian) 该模型的主干是ResNet。 在我们的培训中,我们使用CIFAR10作为数据集。 # To train with Se python train_CIFAR10.py --prefix 4 --device 1 --epoch 160 --att_type se # To trian with CBAM python train_CIFAR10.py --prefix 5 --device 1 --epoch 160 --att_type cbam 验证结果 ResNet50(训练了160个时代)ACC@1=93.41% A
资源详情
资源评论
资源推荐
收起资源包目录




















共 16 条
- 1





















每天痛苦与更好的
- 粉丝: 25
- 资源: 4539
上传资源 快速赚钱
我的内容管理 收起
我的资源 快来上传第一个资源
我的收益
登录查看自己的收益我的积分 登录查看自己的积分
我的C币 登录后查看C币余额
我的收藏
我的下载
下载帮助


会员权益专享
安全验证
文档复制为VIP权益,开通VIP直接复制

评论2