Grad_Cam-pytorch-resnet50

Class activate map Train dog and cat dataset by resnet50
Paper:https://arxiv.org/pdf/1610.02391v1.pdf

Install

python3.6  
pytorch1.2   
cuda 10.0  

Dataset

Dog and Cat
Dataset download link
Pre_model link Extraction code:0uqz

Usage

train model

python3 run.py

generate grad_cam image

python3  test_grad_cam.py /home/xxx_model.pth /home/your_test_images.jpg  

Result