Reference implementation of saliency map (aka attribution map) techniques for deep learning in Python 3.6. Uses MNIST and LeNet 5.
Run
git clone https://github.com/andrewschreiber/numpy-saliency.git
cd numpy-saliency
pip3 install -r requirements.txt
python3 main.py
You can train the model, test the model (using pretrained weights), and/or generate a saliency map via uncommenting the code.
Vanilla Gradients
Integrated Gradients
Adversarial Saliency maps