Any gradcam / heatmap code available for v2?

Has anyone implemented gradcam or cam++ in v2 yet? Or even just basic heatmaps?

I’m missing having those for debugging but wanted to check before I work on it.

Thanks!

you may check the course nb pets more … https://github.com/fastai/fastai2/blob/master/nbs/course/lesson6-pets-more.ipynb

2 Likes

Perfect thanks for the link @s.s.o!

There’s another (IMO cleaner) implementation in this chapter in the book: https://github.com/fastai/fastbook/blob/master/18_CAM.ipynb

1 Like

I created a package that does this too, though it’s not been tested with multi-label models

Repo: https://github.com/Synopsis/fastai2_extensions
Forum Post: Fastai-extensions Package

1 Like