背景

关于神经网络可视化工具，一直比较喜欢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