search

yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...
Apache License 2.0
729 stars 167 forks source link

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

Open ghost opened 3 years ago

ghost commented 3 years ago

您好，非常感谢您提供的关于目标检测方面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

zshanggu commented 2 years ago

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

IMoonKeyBoy commented 2 years ago

解决了没？我也想可视化中间一层，但是一直不成功；