CBAM.PyTorch icon indicating copy to clipboard operation
CBAM.PyTorch copied to clipboard

Non-official implement of Paper:CBAM: Convolutional Block Attention Module

Results 10 CBAM.PyTorch issues
Sort by recently updated
recently updated
newest added

Hello, may you provide a pre-trained model of cbam, thank you.

For the file resnet_cbam.py, I think line31 and line 33 are not consistent with the paper. fc1 and fc2 should be nn.Linear because the paper said: > Both descriptors are...

我将模块加在resnet网络上,训练时出现max_out的print结果为两个tensor,被认定为是list,无法运行,请问大家有遇到此类问题吗

hi I ma getting this error while testing Traceback (most recent call last): File "test.py", line 213, in test_and_generate_result_round2('2','resnet50',416, True) File "test.py", line 133, in test_and_generate_result_round2 my_model = resnet50.MyResNet50(model_ft) NameError:...

I find your code can't reach the accuary

After applying Channel Attention Module, maybe it would be better to apply a convolution layer in order to modify the channels to the original value (usually 3 channels), instead of...