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: