使用PlotNeuralNet绘制自己的网络结构图

绘制效果:
使用PlotNeuralNet绘制自己的网络结构图_第1张图片

源码下载:
PlotNeuralNet

只依赖于python和Latex,在ubuntu16.04下只需要安装texlive-latex-extra就行:

sudo apt-get install texlive-latex-extra

需要注意的是,如果bash中默认的python版本是python2,就需要将tikzmake.sh中的

python $1.py 

修改成

python3 $1.py 

然后就可以正常使用了。

这个工具其实就是用python对latex的绘图脚本更进一步进行封装,将常见的网络结构图形封装成python函数,以简化绘制过程。
比较遗憾的是,作者虽然在“examples”中给出了一些网络的示例,但是仅有.tex版本,而未提供python脚本,而“pyexample”中仅有两个简单的例子,很多函数并没有说明怎么绘制,所以需要自己花一点时间摸索一下。

如何使用这个工具,这两个博客帮助很大:
PlotNeuralNet简单教程

利用plotneuralnet工具生成卷积神经网络的三维结构图,使用,PlotNeuralNet,立体

所有的函数封装在pycore/tikzeng.py中实现,但是我感觉作者放出来的代码可能并不完整。。。有些示例网络结构(如上面图中第三个,Holistically-Nested Edge Detection )中的连接结构就无法通过作者提供给代码绘制,因为tikzeng.py中使用to_skip这个函数进行网络跨层连接,而这个函数中的箭头起点和终点路线都是固定的,只能从结构的上方进行连接,因此如果要画出示例图案那样从侧面引出的箭头还需要自己改一下代码

你可能感兴趣的:(python,神经网络,深度学习)