search

IBM / action-recognition-pytorch

This is the pytorch implementation of some representative action recognition approaches including I3D, S3D, TSN and TAM.
Apache License 2.0
244 stars 47 forks source link

Grad-CAM #12

Open koyakuwe opened 2 years ago

koyakuwe commented 2 years ago

Does anyone who read this message tell me how to generate Grad-CAM like in the image I attached: image

I have used https://github.com/jacobgil/pytorch-grad-cam framework but my work fails (I can draw heatmap on frames; but every different raw frame has the same heatmap img).

My model is 2D ResNet and input shape is NT x C x H x W. Coded using PyTorch.

Thank you.