Skip to content

Testing Grad-CAM localization ability on brain tumor classification task

License

Notifications You must be signed in to change notification settings

daniele-marini/Grad-CAM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Open In Colab

Grad-CAM

Gradient-weighted Class Activation Mapping (Grad-CAM) is a class-discriminative localization technique that can generate visual explanations from any CNN-based network without any architectural requirement or re-training. This approach uses the gradient flowing into the final convolutional layer to produce a localization map highlighting the important regions in the image for predicting the specific target.

Task Description

The aim of the project it to apply Grad-CAM technique to a CNN trained on a brain tumor classification task and evaluate the visual explanation.

Model and Dataset

The model used for the classification is ResNet50, which has been fine tuned by using the "Brain MRI images for brain tumor detection" dataset.

Results

The results of the experiment were highly satisfactory, demonstrating a remarkable ability to accurately localize the tumor's exact region as we can see in figure below.

image

About

Testing Grad-CAM localization ability on brain tumor classification task

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published