Skip to content

Latest commit

 

History

History
32 lines (26 loc) · 1.54 KB

README.md

File metadata and controls

32 lines (26 loc) · 1.54 KB
Raw

Salient-Relevance-Propagation

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.

Prerequisites

  • 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

Citation

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}
}