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

Using Grad-CAM for SSD network

Open Fpooyan opened this issue 5 years ago • 3 comments

Hi, Thanks for publishing this great repo! I am debugging an SSD network used for a 3-class object detection task. Can I use this repo for visualizing such a network as well? I would appreciate it if you could provide me the initial steps...

Fpooyan avatar May 19 '20 13:05 Fpooyan

@Fpooyan 您好,对于目标检测每个网络都不一样,SSD网络生成CAM图与RetinaNet比较类似,请问您基于哪个工程做SSD网络的可视化?可以互相交流一下~

yizt avatar May 19 '20 14:05 yizt

Thanks for your quick response! I do not speak Chinese, so I am using translate to get you. I want to find a way to visualize the base layers of my SSD so that I can see what features it is concentrating on. As SSD networks work with prior boxes, I thought their Grad-CAM must be something different.

Fpooyan avatar May 20 '20 06:05 Fpooyan

@Fpooyan 您好,prior boxes 就是anchors;ssd,retinanet和faster r-cnn都是anchor-based 的目标检测;SSD和retinanet都是one-stage网络,都有特征金字塔,所有个人认为它们的Grad-CAM是类似的。请问你想基于哪个工程来做?

yizt avatar May 21 '20 14:05 yizt