生成神经网络体系结构图像
neuralnet-visualize的Python项目详细描述
神经网络可视化器
一般说明
用给定的结构创建神经网络图像的模块。与模型摘要相比,这是一个方便的工具,可以查看您的网络是如何构建的。在
安装
安装前
在安装模块之前,在您的提示下运行以下命令来安装graphviz
$ sudo apt install graphviz
正常安装
^{pr2}$开发安装
$ git clone https://github.com/AnuragAnalog/nn_visualize.git
$ cd nn_visualize
安装后
安装模块后,如果要升级模块,请运行以下命令。在
sudo pip3 install neuralnet-visualize --upgrade
未来工作
- [x] 添加卷积层、最大池、平坦层
- []添加序列模型层
- []直接从pickle文件
- []激活功能的特定颜色
- []直接从Pythorch模型转换
神经网络可视化工具版本历史
0.2.3
- 从_pytorch方法添加
- 修复了一个图像中多个标题的错误
- 改变了from_tensorflow方法的逻辑
- 修正了一个错误,不可训练图层的唯一图层名
0.2.2
- 添加了更多docstring
- 现在可以在py2中下载了
0.2.1
- 为图像添加了标题
- 优化了一些代码
0.2.0款
- 使它独立于张量流
- 修正了一些错误
0.1.4
- 修正了一些错误
- 添加了更多的层maxpooling,avgpooling,flatten
- 更改了conv2d层的形状
0.1.3
- 添加了类docstrings和示例
- 为转换层增加了一些参数
- 在from_tensorflow调用之后禁用了层的自定义添加
0.1.2
- 添加了Conv层,带有kernel size参数
- 重构的添加层函数
- 将Conv层添加到from_tensorflow
0.1.1
- 添加了一些可能的例外情况
- 在初始导入时重新构造了基本函数
0.1.0款
- 添加的文档
0.0.4
- 重构的汇总函数
- 添加了一个代码部分,它将输出层涂成红色
- 从_tensorflow方法添加
0.0.3
- 添加了更多显示方向
- 重构构建网络并添加层代码
- 添加了一些可以保存图像的文件扩展名
- 增加汇总功能
0.0.2
- 为图层类型添加颜色
- 增加密集层
0.0.1
初始版本
- 起始代码
- 项目
标签: