search

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

How can I implement Grad-CAM on object detection?? #5

Closed sihan-shao closed 4 years ago

sihan-shao commented 4 years ago

I find that almost code about Grad-CAM to be aimed classification model, so can it be implement on object detection model??

Recently I read some paper about attention, these papers use Grad-CAM to compare the performance improvement of object detection model, but I can find the code about visualization.

Can you give me some suggestion for modifying the code ?

Thanks!!!

yizt commented 4 years ago

@SHAOSIHAN 您好，我会尽快实现目标检测部分，完成后通知您

yizt commented 4 years ago

@SHAOSIHAN 您好，已经更新目标检测部分，请更新工程！