caffe基础-11绘制网络

来源:互联网 发布:淘宝嘉定仓库 编辑:程序博客网 时间:2024/05/20 08:26

caffe基础-11绘制网络

1、安装graphViz:# sudo apt-get install graphviz2、安装pip(if need):参考:https://zhidao.baidu.com/question/1240554972151163659.html3 、安装pydot:# sudo pip install pydot4、绘图命令: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。

图片参考

这里写图片描述