CBAM.PyTorch
CBAM.PyTorch copied to clipboard
Non-official implement of Paper:CBAM: Convolutional Block Attention Module
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...
datasets?