├── .gitignore ├── README.md ├── main.py ├── main_EfficientNet.py ├── models ├── BA_effcientnet.py ├── BA_mobilenetv3.py ├── BA_module.py ├── BA_resnet.py ├── BA_resnet_fca.py ├── BA_resnext.py ├── DCT_extration.py ├── __init__.py ├── mobilenetv3.py ├── se_module.py ├── se_resnet.py └── utils.py ├── results ├── ba_efficientnet-b0 │ └── metric.txt ├── ba_mobilenetv3-large │ ├── Loss_plot.txt │ ├── lr_plot.txt │ ├── train_prec1.txt │ ├── train_prec5.txt │ ├── val_prec1.txt │ └── val_prec5.txt ├── ba_mobilenetv3-small │ ├── Loss_plot.txt │ ├── lr_plot.txt │ ├── train_prec1.txt │ ├── train_prec5.txt │ ├── val_prec1.txt │ └── val_prec5.txt ├── ba_resnet101 │ ├── Loss_plot.txt │ ├── lr_plot.txt │ ├── train_prec1.txt │ ├── train_prec5.txt │ ├── val_prec1.txt │ └── val_prec5.txt ├── ba_resnet50 │ ├── Loss_plot.txt │ ├── lr_plot.txt │ ├── train_prec1.txt │ ├── train_prec5.txt │ ├── val_prec1.txt │ └── val_prec5.txt └── ba_resnext50 │ ├── Loss_plot.txt │ ├── lr_plot.txt │ ├── train_prec1.txt │ ├── train_prec5.txt │ ├── val_prec1.txt │ └── val_prec5.txt └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/main.py -------------------------------------------------------------------------------- /main_EfficientNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/main_EfficientNet.py -------------------------------------------------------------------------------- /models/BA_effcientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_effcientnet.py -------------------------------------------------------------------------------- /models/BA_mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_mobilenetv3.py -------------------------------------------------------------------------------- /models/BA_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_module.py -------------------------------------------------------------------------------- /models/BA_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_resnet.py -------------------------------------------------------------------------------- /models/BA_resnet_fca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_resnet_fca.py -------------------------------------------------------------------------------- /models/BA_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/BA_resnext.py -------------------------------------------------------------------------------- /models/DCT_extration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/DCT_extration.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/mobilenetv3.py -------------------------------------------------------------------------------- /models/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/se_module.py -------------------------------------------------------------------------------- /models/se_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/se_resnet.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/models/utils.py -------------------------------------------------------------------------------- /results/ba_efficientnet-b0/metric.txt: -------------------------------------------------------------------------------- 1 | TOP-1 71.704 2 | TOP-5 90.206 -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/Loss_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/Loss_plot.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/lr_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/lr_plot.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/train_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/train_prec1.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/train_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/train_prec5.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/val_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/val_prec1.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-large/val_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-large/val_prec5.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/Loss_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/Loss_plot.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/lr_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/lr_plot.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/train_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/train_prec1.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/train_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/train_prec5.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/val_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/val_prec1.txt -------------------------------------------------------------------------------- /results/ba_mobilenetv3-small/val_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_mobilenetv3-small/val_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnet101/Loss_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/Loss_plot.txt -------------------------------------------------------------------------------- /results/ba_resnet101/lr_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/lr_plot.txt -------------------------------------------------------------------------------- /results/ba_resnet101/train_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/train_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnet101/train_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/train_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnet101/val_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/val_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnet101/val_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet101/val_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnet50/Loss_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/Loss_plot.txt -------------------------------------------------------------------------------- /results/ba_resnet50/lr_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/lr_plot.txt -------------------------------------------------------------------------------- /results/ba_resnet50/train_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/train_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnet50/train_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/train_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnet50/val_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/val_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnet50/val_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnet50/val_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnext50/Loss_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/Loss_plot.txt -------------------------------------------------------------------------------- /results/ba_resnext50/lr_plot.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/lr_plot.txt -------------------------------------------------------------------------------- /results/ba_resnext50/train_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/train_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnext50/train_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/train_prec5.txt -------------------------------------------------------------------------------- /results/ba_resnext50/val_prec1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/val_prec1.txt -------------------------------------------------------------------------------- /results/ba_resnext50/val_prec5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/results/ba_resnext50/val_prec5.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoy376/Bridge-Attention/HEAD/utils.py --------------------------------------------------------------------------------