TianhongDai/integrated-gradient-pytorch

Stars
180
Rank 213,097 (Top 5 %)
Language
Python
License
MIT License
Created about 6 years ago
Updated over 2 years ago

TianhongDai/integrated-gradient-pytorch

TianhongDai

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

This is the pytorch implementation of the paper - Axiomatic Attribution for Deep Networks.

Integrated Gradients

This is the pytorch implementation of "Axiomatic Attribution for Deep Networks". The original tensorflow version could be found here.

Acknowledgement

ankurtaly's tensorflow version

Requirements

python-3.5.2
pytorch-0.4.1
opencv-python

TODO List

add more functions as the original code.
finetune the results, make them close to the original paper.

Instructions

Highly recommend to use GPU to accelerate the computation. If you use CPU, I will recommend to select some small networks, such as resnet18. You also need to put your images under examples/.

Lists of networks that support (of course, you can add any networks by yourself)

inception
resnet18
resnet152
vgg19

Run the code

python main.py --cuda --model-type='inception' --img='01.jpg'

Results

Results are slightly different from the original paper, it may have some bugs or need to do some adjustments. I will keep updating it, any contributions are welcome!

Inception-v3

ResNet-18

ResNet-152

VGG-19

reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

hindsight-experience-replay

This is the pytorch implementation of Hindsight Experience Replay (HER) - Experiment on all fetch robotic environments.

mosse-object-tracking

This is the implementation of MOSSE tracking algorithm (correlation filter based).

self-imitation-learning-pytorch

This is the pytorch implementation of ICML 2018 paper - Self-Imitation Learning.

distributed-ppo

This is an pytorch implementation of Distributed Proximal Policy Optimization(DPPO).

google-football-pytorch

It's the pytorch implementation of google research football.

metaworld-sac

div-hindsight

This is the official code of our paper "Diversity-based Trajectory and Goal Selection with Hindsight Experience Relay" [PRICAI 2021].

esil-hindsight

This is the official code of our paper "Episodic Self-Imitation Learning with Hindsight" [Electronics 2020].

wavelet-hdr

This is the official code for our paper "Wavelet-Based Network For High Dynamic Range Imaging" [CVIU 2023].

deep-hdr-baselines

react2-code

This is the official code of our paper "Machine Learning to Support Visual Auditing of Home-based Lateral Flow Immunoassay Self-Test Results for SARS-CoV-2 Antibodies" [Communications Medicine 2022].

dockerfiles

It contains the dockerfiles for the purpose of machine learning / deep learning research.

phd-thesis

daim-rl

This is the official code of our paper "Diversity-Augmented Intrinsic Motivation for Deep Reinforcement Learning" [Neurocomputing 2021].

DouTu