caffe-master/python/draw_net.py 实现绘制caffe中定义的网络模型功能,将.prototxt文件可视化。


需要先安装pydot和protobuf工具


通过Anaconda安装pydot和protobuf工具:

sduo chmod 777 -R ~/anaconda2
conda install protobuf
conda install pydot




如果还缺少其他工具,也可以通过这种方式安装。



绘制Lenet网络模型


MNIST手写体网络模型Lenet:

cd ~/caffe-master
./python/draw_net.py ./examples/mnist/lenet_train_test.prototxt ./python/lenet.png




cifar:

cd ~/caffe-master
./python/draw_net.py ./examples/cifar10/cifar10_quick_train_test.prototxt ./python/cifar.png




更改网络模型方向显示


draw_net.py带三个参数:

  • 第一个参数:网络模型的prototxt文件
  • 第二个参数:保存的图片路径及名字
  • 第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。

更改绘制方向为从下到上:

Lenet:





cifar:


posted on 2017-07-18 21:40  未雨愁眸  阅读(218)  评论(0编辑  收藏  举报