ResNet-50-CBAM-PyTorch
ResNet-50-CBAM-PyTorch copied to clipboard
Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https://www.kaggle.com/puneet6060/intel-image-classification.
Can the author post it again?
How did you draw your heatmap?I can't find the code to draw the heatmap.
The link for pre-trained weights cannot be opened. Can you upload the new file? Thank you very much!
How I run on my own computer, not on the server.
https://github.com/Ugenteraan/ResNet-50-CBAM-PyTorch/blob/fc43294d423050ff14da8016fd6948b99ad4a912/load_dataset.py#L55