Pytorch c++ Grad-CAM

C++
1

Hi,
Does someone already try to do Grad-CAM (heatmap) with pytorch c++?

For now, I did the training in python and load the model with:

m_module = torch::jit::load(pathscript);

I’m pretty sure that I can’t use jit trace to do Grad-CAM and I will need to redefine my CNN in c++ directly.

struct Net: torch::nn::Module

Can someone confirm?

Thanks

2 Likes
2

I am very curious about this as well. Any luck?

3

Are there any news on this thread? it is still current