ResNet-50-CBAM-PyTorch icon indicating copy to clipboard operation
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.

Results 8 ResNet-50-CBAM-PyTorch issues
Sort by recently updated
recently updated
newest added

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