Neural network visualization toolkit for tf.keras
Project description
tf-keras-vis
tf-keras-vis is a visualization toolkit for debugging Keras models with Tensorflow2, but not original Keras.
The features of tf-keras-vis are based on keras-vis, but tf-keras-vis's APIs doesn't have compatibility with keras-vis's because, instead of getting it, we prioritized to get following features.
- Support processing multipul images at a time as a batch
- Support tf.keras.Model that has multipul inputs (and, of course, multipul outpus too)
- Allow to use optimizers that embeded in tf.keras
And then we will add some algorisms such as below.
- SmoothGrad: removing noise by adding noise (DONE)
- Deep Dream
- Style transfer
Requirements
- Python 3.5, 3.6, 3.7 or 3.8
- tensorflow>=2.0.0
Installation
- PyPI
$ pip install tf-keras-vis tensorflow
- Docker
$ docker pull keisen/tf-keras-vis:0.2.4
You can find other images (that's nvidia-docker images) at dockerhub.
Usage
Please see examples/attentions.ipynb, examples/visualize_dense_layer.ipynb and examples/visualize_conv_filters.ipynb.
- Run Jupyter notebooks on Docker
$ docker run -itd -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.2.4
Or, if you have GPU processors,
$ docker run -itd --runtime=nvidia -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.2.4-gpu
API Documentation
T.B.D.
Known Issues
- With InceptionV3, ActivationMaximization doesn't work well, that's, it might generate meanninglessly bulr image.
- With cascading model, Gradcam doesn't work well, that's, it might occur some error.
- Unsupported
channels-first
models and datas.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tf-keras-vis-0.2.5.tar.gz
(10.9 kB
view hashes)
Built Distribution
Close
Hashes for tf_keras_vis-0.2.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 49b272f85a26230a75c170f976fa87a9d10427638245754174ba813da15f20c3 |
|
MD5 | f07522ab2c5ac7d919c03c7156e7d32e |
|
BLAKE2b-256 | 3a1bb858c3cd7cea0da6219d0badaef55d7b29e0d72f55df791d8e9aebc85f79 |