-
train the classification network
python scripts/train_cls.py --img_dir your_img_dir
-
generate cam using trained classification network
python scripts/cam.py --checkpoint best_checkpoint.pth
Date | Paper source | Title | Code |
---|---|---|---|
2020.06 | CVPR workshop | Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networks | PyTorch |
2018.05 | WACV | Grad-CAM++: Generalized Gradient-based Visual Explanations for Deep Convolutional Networks | |
2017.10 | ICCV | Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization | PyTorch |
2016.06 | CVPR | Learning Deep Features for Discriminative Localization | PyTorch |
- Part of the code is adapted from open-source codebase and original implementations of algorithms, we thank these authors for their fantastic and efficient codebase, such as DRS and pytorch-grad-cam.