HYUNMIN-HWANG / AI604-Computer-Vision-Project

AI604 Computer Vision Project "Weakly Supervised Generation of Fine-Grained Segmentation Labels via Coarse Class Activation Maps"

AI604-Computer-Vision-Project

AI604 Computer Vision Project "Weakly Supervised Generation of Fine-Grained Segmentation Labels via Coarse Class Activation Maps"

Generating gradCAM heatmaps

Code was edited from https://github.com/jacobgil/pytorch-grad-cam

You can install gradCAM with pip install grad-cam.

Then replace the original cam.py file with our version and add utils.py.