Caffe中网络的绘制

1、安装graphviz:

 sudo apt-get install graphviz  #依赖库

2、安装pydot

sudo pip install pydot

3、使用命令行进行网络的绘制

#第一个参数是Caffe文件中Python文件夹下的draw_net.py文件
#第一个参数:网络模型的prototxt文件
#第二个参数:保存图像的路径及名称
#第三个参数:--rankdir=x,y 有四种选项,分别是LR,RL,TB,BT,分别是从左到右,从右到左,从上到下,从下到上。默认为LR

sudo python /home/pcb/caffe/python/draw_net.py /home/pcb/caffe/examples/mnist/lenet_train_test.prototxt /home/pcb/examples/mnist/lenet.png --rankdir=LR

这样绘制好的网络模型就存放在设定的路径下了!

问题:
1 ImportError: No model named skimage.io

解决办法:sudo pip install scikit-image

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值