PyTorch implementation of Layer-wise Relevance Propagation based on this tutorial, which is used in this paper Beyond Saliency: Understanding Convolutional Neural Networks from Saliency Prediction on Layer-Wise Relevance Propagation.
We also applies context-aware saliency detection which is available here.
- Python 3.x
- PyTorch
- Jupyter Notebook
- pre-trained VGG-Face model
PyTorch does not provide a pre-trained VGG-Face model. But you can download a Torch model from here and then convert it to a PyTorch model using this tool. - Large Scale Visual Recognition Challenge 2012 (ILSVRC2012) validation dataset
If you use this code for your research, please cite our paper Beyond Saliency: Understanding Convolutional Neural Networks from Saliency Prediction on Layer-Wise Relevance Propagation
@article{li2019beyond,
title={Beyond saliency: understanding convolutional neural networks from saliency prediction on layer-wise relevance propagation},
author={Li, Heyi and Tian, Yunke and Mueller, Klaus and Chen, Xin},
journal={Image and Vision Computing},
year={2019},
publisher={Elsevier}
}