caffe基础-11绘制网络
1、安装graphViz:
# sudo apt-get install graphviz
2、安装pip(if need):
参考:https://zhidao.baidu.com/question/1240554972151163659.html
3 、安装pydot:
# sudo pip install pydot
4、绘图命令:
sudo python /home/terrence/caffe/python/draw_net.py /home/terrence/caffe/examples/mnist/lenet_train_test.prototxt /home/terrence/caffe_case/lenet.png --rankdir=LR
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第三个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。