shirgur / AGFVisualization

Official code for "Attribution Guided Factorization for Neural Networks Visualization"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AGF-Visualization

Official code for "Visualization of Supervised and Self-Supervised Neural Networks via Attribution Guided Factorization"

Baseline methods are taken from link, and link.

Prerequisites

  • python 3.6
  • pytorch 1.1
  • torchvision
  • numpy
  • scipy
  • scikit-learn
  • scikit-image
  • Pillow
  • matplotlib
  • OpenCV
  • tqdm
  • h5py
  • imageio
  • tensorboard

About

Official code for "Attribution Guided Factorization for Neural Networks Visualization"

License:MIT License


Languages

Language:Python 53.2%Language:Jupyter Notebook 46.8%