文件名称:Python-BAM和CBAM的官方PyTorch代码
文件大小:9KB
文件格式:ZIP
更新时间:2022-09-01 02:32:35
Python开发-机器学习
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
【文件预览】:
attention-module-master
----train_imagenet.py(11KB)
----scripts()
--------train_imagenet_resnet50_cbam.sh(216B)
--------train_imagenet_resnet50_bam.sh(214B)
----MODELS()
--------cbam.py(4KB)
--------model_resnet.py(7KB)
--------bam.py(3KB)
----README.md(2KB)