项目作者： daixiangzi

项目描述：
Class activate map

高级语言： Python

项目主页：

项目地址: git://github.com/daixiangzi/Grad_Cam-pytorch-resnet50.git

创建时间： 2020-03-05T06:43:44Z
项目社区：https://github.com/daixiangzi/Grad_Cam-pytorch-resnet50
开源协议：Apache License 2.0
下载

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

both dog cat