Skip to content

About

A simple implementation of GradCAM (Gradient-weighted Class Activation Mapping) using PyTorch and OpenCV.

Topics

Resources

Stars

6 stars

Watchers

1 watching

Forks

Latest commit

 

History

3 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GradCAM in PyTorch

A simple implementation of GradCAM (Gradient-weighted Class Activation Mapping) using PyTorch and OpenCV. The provided code applies Grad-CAM to a MobileNetV2 model trained on the ImageNet dataset, demonstrating how to generate and overlay heatmaps on input images.

Heatmap with GradCAM

About

A simple implementation of GradCAM (Gradient-weighted Class Activation Mapping) using PyTorch and OpenCV.

Topics

Resources

Stars

6 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages