search

jacobgil / pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
https://jacobgil.github.io/pytorch-gradcam-book
MIT License
10.31k stars 1.54k forks source link

Support CAM in MMDet #249

Open hhaAndroid opened 2 years ago

hhaAndroid commented 2 years ago

Many thanks to the author for the repo. I've created a PR to support mmdet. Welcome those who have time to improve. #https://github.com/open-mmlab/mmdetection/pull/7987

jacobgil commented 2 years ago

This is really great and will bring CAM in object detection to a large audience. Do you think there is a way of also getting gradients there for GradCAM (and the rest of the gradient methods) ?

hhaAndroid commented 2 years ago

@jacobgil Yes. I don't think it's a problem.

hhaAndroid commented 2 years ago

@jacobgil I just briefly verified that gradcam can be supported. It will be updated to PR in the future, welcome to pay attention.

hhaAndroid commented 2 years ago

@jacobgil I already support grad_base, can you please help to review it? Thank you very much.