This is the pytorch implementation of "Axiomatic Attribution for Deep Networks".
- [ ] add more functions as the original code.
- [ ] finetune the results, make them close to the original paper.
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
Lists of networks that support (of course, you can add any networks by yourself)
Run the code
python main.py --cuda --model-type='inception' --img='01.jpg'
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!