Skip to content

christophM/explain-ml

Folders and files

NameName
Last commit message
Last commit date

Latest commit

9c7868f · Jan 20, 2017

History

4 Commits
Jan 20, 2017
Jan 20, 2017
Jan 20, 2017
Jan 20, 2017
Jan 20, 2017
Jan 20, 2017
Jan 20, 2017

Repository files navigation

Explain output of image classifier

VGG-16 [1] trained on ILSVRC-2014 data [2]. The algorithm used for identifying the sub-images that gets the highest probability for the top class is inspired by [3].

Weights for VGG16 can be downloaded here: vgg16_weights.h

Usage

python explain.py -i dog.jpg -m vgg16_weights.h5 -o out.jpg

Input image:

Image of dog playing guitar

Output image:

Image of dog playing guitar, only showing dog

[1] Very Deep Convolutional Networks for Large-Scale Image Recognition K. Simonyan, A. Zisserman arXiv:1409.1556

[2] http://image-net.org/challenges/LSVRC/2014/

[3] "Why Should I Trust You?": Explaining the Predictions of Any Classifier Marco Tulio Ribeiro, Sameer Singh, Carlos Guestrin arXiv:1602.04938

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages