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

25 篇文章 0 订阅
    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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值