kazuto1011 / grad-cam-pytorch

PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I load & run my own trained convnet model in your grad-cam?

bemoregt opened this issue · comments

Hi, @kazuto1011

How can I load & run my own trained convnet model in your grad-cam?

My convnet(CNN) is a 34-layer's resnet, trained using my image datasets.

Thanks in advance.

form @bemoregt.

You can find the solution here #3 . Basically, you only have to replace the model instance.

Hi, @kazuto1011

Thanks a lot.

I solved It.