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

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

python3 run.py

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

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
example		example
utils		utils
LICENSE		LICENSE
MyDataset.py		MyDataset.py
README.md		README.md
config.py		config.py
run.py		run.py
test_grad_cam.py		test_grad_cam.py