caffe学习:绘制网络模型
在确保caffe的Python接口编译正确的情况下
1 安装GraphViz
sudo apt-get install GraphViz
2 安装pydot
sudo pip install pydot
安装好后就可以调用caffe中Python文件夹下的draw_net.py脚本画图了
比如绘制caffe自带的LeNet网络模型
在caffe文件夹下输入:
sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt lenet.png --rankdir=LR
说明:draw_net.py执行的时候带三个三数
第一个参数:网络模型文件
第二个参数:保存的图片的名字
第三个参数:--rankdir=LR或RL/TB/BT 用来表示网络的方向