面向keras的神经网络可视化工具
keras-mri的Python项目详细描述
Keras MRI
keras-mri是一种用于keras的神经网络可视化工具。这仍处于测试阶段,一些网络(如RNN和3D CNN)和层类型尚不受支持。
安装
pip install keras-mri
用法
importkmrifromkeras.modelsimportSequentialfromkeras.layersimportDensemodel=Sequential([Dense(8,input_shape=(3,))])input_batch=[[1,2,3],[7,8,9]]kmri.visualize_model(model,input_batch)
运行后,导航到http://127.0.0.1:5000/
示例
examples文件夹包含演示keras mri用法的模型。要运行其中一个示例,请从其中一个示例文件夹中运行以下命令
pip install -r requirements.txt python main.py
简单的强化学习模型,学会了在chrome中玩恐龙游戏。演示如何可视化具有多个输入的卷积神经网络。这是基于chrome-dino-ai。
演示如何使用keras mri可视化标准图像分类器(vgg、mobilenet和resnet)。
开发
在本地安装软件包
pip install -e .
deploy
# increase version number in setup.py
# Move unrelease changes to next version in changelog.md
# commit, tag with vx.x.x, and push
# Delete build, dist, and egg-info
python setup.py sdist bdist_wheel
python -m twine upload dist/*