Grad-CAM.pytorch icon indicating copy to clipboard operation
Grad-CAM.pytorch copied to clipboard

gradient = self.gradient[proposal_idx].cpu().data.numpy()

Open ghost opened this issue 4 years ago • 2 comments

您好,非常感谢您提供的关于目标检测方面CAM的代码,但是其中有一处错误我一直无法解决

grad_cam.py:

grad = self.gradient[proposal_idx].cpu().date.numpy() 我在detectron2框架中进行检测的grad-cam输出时,其中我的self.gradient的维度为[1, C, W, H],所以self.gradient[proposal_idx]会报错。但是我看其他的issues中,您说self.gradient的维度应该为[num_proposals,C,H,W]。我不知道的哪里出现了问题。我的layer_name = backbone.bottom_pu.res5.2

ghost avatar Oct 22 '21 03:10 ghost

Hi, have you resolved this issue? I meet this problem, too.

zshanggu avatar Jul 27 '22 08:07 zshanggu

解决了没?我也想可视化中间一层,但是一直不成功;

IMoonKeyBoy avatar Nov 01 '22 07:11 IMoonKeyBoy