diff --git a/README.md b/README.md index b015db8..ba2a73b 100644 --- a/README.md +++ b/README.md @@ -1,12 +1,16 @@ # Saliency Methods -## Introduction +## Overview This repository contains code for the following saliency techniques: -* XRAI -* SmoothGrad -* Vanilla Gradients -* Guided Backpropogation -* Integrated Gradients -* (Guided) Grad-CAM +### 1. Vanilla Gradients +![gradient](images/gradient.png) +### 2. Guided Backpropogation (with Smoothgrad) +![guided](images/guided_backrop_smoothgrad.png) +### 3. Integrated Gradients +![integrated](images/integrated_gradient.png) +### 4. (Guided) Grad-CAM +![gradcam](images/guided_grad_cam.png) +### 5. XRAI +![xrai](images/xrai.png) ## Remarks The methods should work with all models from the [torchvision](https://github.com/pytorch/vision) package. Tested models so far are: