1、安装必要的库。
$ sudo apt-get install GraphViz
$ sudo pip install pydot
2、绘制网络
参数说明:
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
例子:
$sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt net.png --rankdir=BT
3、绘制自己的网络模型
未完待续。。。