https://github.com/poloclub/interactive-classification
Interactive Classification for Deep Learning Interpretation
https://github.com/poloclub/interactive-classification
deep-learning inpainting neural-networks visualization
Last synced: 5 months ago
JSON representation
Interactive Classification for Deep Learning Interpretation
- Host: GitHub
- URL: https://github.com/poloclub/interactive-classification
- Owner: poloclub
- License: mit
- Created: 2018-02-13T02:12:51.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2023-01-25T07:47:14.000Z (over 2 years ago)
- Last Synced: 2025-05-06T18:14:45.288Z (5 months ago)
- Topics: deep-learning, inpainting, neural-networks, visualization
- Language: JavaScript
- Homepage: https://cabreraalex.com/#/paper/interactive-classification
- Size: 11.7 MB
- Stars: 77
- Watchers: 7
- Forks: 17
- Open Issues: 8
-
Metadata Files:
- Readme: README.md
- License: LICENSE.md
Awesome Lists containing this project
README
# Interactive Classification for Deep Learning Interpretation
We have designed and developed an interactive system that allows users to experiment with deep learning image classifiers and explore their robustness and sensitivity.
Selected areas of an image can be removed in real time with classical computer vision inpainting algorithms, allowing users to ask a variety of "what if" questions by experimentally modifying images and seeing how the deep learning model reacts.
The system also computes class activation maps for any selected class, which highlight the important semantic regions of an image the model uses for classification.
The system runs fully in browser using Tensorflow.js, React, and SqueezeNet.
An advanced inpainting version is also available using a server running the PatchMatch algorithm from the [GIMP Resynthesizer plugin](https://github.com/bootchk/resynthesizer).
[](https://youtu.be/llub5GcOF6w)
This is the code repository for the accepted [CVPR 2018][cvpr18] Demo: **Interactive Classification for Deep Learning Interpretation**.
Visit our research group homepage [Polo Club of Data Science][poloclub] at [Georgia Tech](http://www.gatech.edu) for more related research!
## Example Scenario: Interpreting "Failed" Classification
The modified image (left), originally classified as *dock* is misclassified as *ocean liner* when the masts of a couple boats are removed from the original image (right).
The top five classification scores are tabulated underneath each image.
## Installation
Download or clone this repository:
```bash
git clone https://github.com/poloclub/interactive-classification.git
```
Within the cloned repo, install the required packages with yarn:
```bash
yarn
```
## Usage
To run, type:
```bash
yarn start
```
## Advanced Inpainting
The following steps are needed to set up PatchMatch inpainting, which currently only works on Linux:
1. Clone the [Resynthesizer](https://github.com/bootchk/resynthesizer) repository and follow the instructions for building the project (stop after running `make`)
2. Find the `libresynthesizer.a` shared library in the generated `lib` folder and copy it to the `inpaint` folder in this repository
3. Run `gcc resynth.c -L. -lresynthesizer -lm -lglib-2.0 -o prog` (may have to install glib2.0 first) to generate the `prog` executable
4. You can now run `python3 inpaint_server.py` and PatchMatch will be used as the inpainting algorithm when running the React application with `yarn start`.
## Citation
**[Interactive Classification for Deep Learning Interpretation][site]**
[Angel Cabrera][angel], [Fred Hohman][fred], [Jason Lin][jason], [Duen Horng (Polo) Chau][polo]
*Demo, Conference on Computer Vision and Pattern Recognition (CVPR). June 18, 2018. Salt Lake City, USA.*
```
@article{cabrera2018interactive,
title={Interactive Classification for Deep Learning Interpretation},
author={Cabrera, Angel and Hohman, Fred and Lin, Jason and Chau, Duen Horng},
journal={Demo, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
year={2018},
organization={IEEE}
}
```
## Researchers
| Name | Affiliation |
|-----------------------|---------------------------------|
| [Angel Cabrera][angel] | Georgia Tech |
| [Fred Hohman][fred] | Georgia Tech |
| [Jason Lin][jason] | Georgia Tech |
| [Duen Horng (Polo) Chau][polo] | Georgia Tech |
## License
MIT License. See [`LICENSE.md`](LICENSE.md).
## Contact
For questions or support [open an issue][issues].
[site]: http://fredhohman.com/papers/interactive-classification
[paper]: http://fredhohman.com/papers/18-interactive-cvpr.pdf
[video]: https://youtu.be/llub5GcOF6w "Video"
[fred]: http://fredhohman.com "Fred Hohman"
[angel]: http://cabreraalex.com/ "Alex Cabrera"
[jason]: http://jlin.xyz/ "Jason Lin"
[polo]: http://www.cc.gatech.edu/~dchau/ "Polo Chau"
[issues]: https://github.com/poloclub/interactive-classification/issues "Issues"
[cvpr18]: http://cvpr2018.thecvf.com
[poloclub]: https://poloclub.github.io