Python + Graphviz绘制神经网络结构图
大家在网上经常能够看到绘制精美的网络节点图形,但这些图形往往有大量节点和连线,如果使用所见即所得的绘图工具(比如微软的visio)往往需要花费大量时间来添加节点,调整布局。Graphviz是款非常好的基于脚本语言
([dot](http://www.graphviz.org/pdf/dotguide.pdf),[中文](http://www.jianshu.com/p/5b02445eca1d))的绘图工具,能很好的避免浪费时间在布局调整上,让人把精力更多的集中在绘图逻辑上。但针对节点较多的网络,不可避免需要投入大量尽量来写重复的脚本代码。
这里,我为大家用python编写了一个简单的dot脚本生成工具(MakeNN),可以很方便的输入参数生成nn结构图。在使用本工具之前,需要确保你的电脑已经正常配置了[python](http://blog.csdn.net/yanzi1225627/article/details/19575437)及[graphviz](http://liyanrui.is-programmer.com/posts/6261.html)开发环境。(windows的python开发环境配置)
python源码下载:https://github.com/rtygbwwwerr/MakeNN1
下图是绘制结果:
对应的python代码如下,可根据实际需求修改:
def