gradient = self.gradient[proposal_idx].cpu().data.numpy() #52
Description
您好，非常感谢您提供的关于目标检测方面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