Caffe(18)——绘制网络模型

来源:互联网 发布:linux spi驱动 编辑:程序博客网 时间:2024/05/16 05:08
    python/draw_net.py。这个文件,就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。    1 安装GraphViz
# sudo apt-get install GraphViz
2 安装pydot
# sudo pip install pydot
安装好后就可以用脚本来绘制图片来draw_net.py执行的时候带三个参数1)网络模型prototxt文件2)保存的图片路径及名字3)--rankdir=x,x有四种选项,分别是LR,RL,TB,BT。用来表示网络方向,从左到右分,从右到左,从上到下,从下到上。默认为LR。例:绘制Lenet 
# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

这里写图片描述
例:绘制cifar10的模型

# sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT
原创粉丝点击