Python-BAM和CBAM的官方PyTorch代码

时间:2022-09-01 02:32:35
【文件属性】:

文件名称: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)

网友评论