Linux下进行
在caffe/python下会有一个draw_net.py用于绘制prototxt格式的模型结构原理图。
1.安装python的依赖包
打开cmd输入以下内容,若网速不行,容易中断,请多次尝试。
sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
2.安装python依赖的画图工具graphviz
sudo apt-get install graphviz
3.切换到root用户
sudo su
4.安装pydot
pip install pydot
5.将/usr/local/lib/python2.7/dist-packages 文件下的内容,全部复制到home/anaconda2/lib/python2.7/site-packages文件。
如下图所示:
6.绘制手写数字的LeNet网模型结构图
在draw_net.py所在的路径caffe/python下打开终端输入:
Python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png –rankdir=TB
按回车键可出现模型结构图如下:
在线呈现
1.直接打开http://ethereon.github.io/netscope/#/editor
2.将prototxt文件复制到左边框里,然后按shift-enter,就可以得到模型结构原理图,如下所示:
两种方法比较如下:
法1绘制出的结构图保存了参数信息,细节更丰富,但是结构不是很清晰明了。
法2绘制出的图直观简洁,易于快速了解网络模型,但是缺少层内的细节信息。