yizt / Grad-CAM.pytorch

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

其他框架下实现 #33

Open Yuesheng321 opened 3 years ago

Yuesheng321 commented 3 years ago

您好,请问可以在mmdetection(https://github.com/open-mmlab/mmdetection)框架下实现retinanet的grad-cam吗?

yizt commented 3 years ago

@Yuesheng321 您好,mmdetection 很少用,不熟悉,理论上都是可以的;您也可以说一下实现的时候有什么问题

zhanghaoyue commented 3 years ago

torchvision 的 RCNN呢?

Frank1o commented 2 years ago

@Yuesheng321 请问您成功了吗?

Frank1o commented 2 years ago

@yizt 在mmdet中使用该方法时,mmdet调用模型是通过config配置文件,您给的方法是通过torchvision调用,然后就失败了。。