jacobgil / pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
https://jacobgil.github.io/pytorch-gradcam-book
MIT License
10.5k stars 1.55k forks source link

Grad-cam for 3D CNN #351

Open koopa31 opened 2 years ago

koopa31 commented 2 years ago

Hi, I've been trying to use your Grad CAM implementaiton on a 3D CNN. My input is a tensor of size (1, 2, 70, 70, 70), but my output is of size (2, 70, 70, 9). I don't understand that 9, do you know why ? Thank you very much

yanhan13944047669 commented 1 year ago

Hi, I've been trying to use your Grad CAM implementaiton on a 3D CNN. My input is a tensor of size (1, 2, 70, 70, 70), but my output is of size (2, 70, 70, 9). I don't understand that 9, do you know why ? Thank you very much

Have you found a solution?

Wajahat0 commented 1 year ago

Did you find the solution?

Rohit8y commented 1 year ago

@jacobgil Can you look into it?