背景
关于神经网络可视化工具,一直比较喜欢netron,恰如其分地满足我对工具的颜控心理:简洁而美观!
且多模型支持,唯一不能满足的也仅仅是不能拖拽和不能生成html代码(特殊要求),也有可能是本人不会用!
于是为了满足本人的特殊需求,简单地写了ndraw,用来可视化神经网络。
支持
- tensorflow的Sequential对象可视化
- tensorflow的Model对象可视化
- h5模型文件可视化
- pb模型文件可视化
- 其他模型暂不支持
- 自定义画图
- 支持样式主题修改
- 支持导出html文件
- 支持生成html字符串
TF模型可视化
import ndraw
ndraw.server("model.h5")
自定义画图
# -*- encoding: utf-8 -*-
import ndraw
from ndraw import Node
graph = ndraw.AutoGraph()
graphs = graph.create("开始")\
.to(Node("过程1",["1.xxx","2.xxx"]))\
.to(Node("过程2",["1.xxx","2.xxx"]))\
.to("结束")
.server()
如何使用
有用的话支持点个Star