Grad-CAM Guided Attention Module for Fine-grained Visual Classification

Code release for Grad-CAM Guided Attention Module for Fine-grained Visual Classification https://arxiv.org/abs/2101.09666

Dataset

• CUB-200-2011
• FGVC-Aircraft
• FGVC-Aircraft

Requirements

• python 3.6
• PyTorch 1.1.0
• torchvision

Training

python cars_sp_CAM.py

Contact

Thanks for your attention! If you have any suggestion or question, you can leave a message here or contact us directly:

• xscaniggia@bupt.edu.cn
• mazhanyu@bupt.edu.cn