用Caffe自带程序画网络结构图

我的系统环境:Win10+VS2013+Anaconda3(Python=3.5)+Caffe
Caffe环境的搭建和配置,这里就不赘述了。懒人办法:可直接去GitHub下载编译好的Caffe程序。

1. 画图工具:

主要使用Caffe自带的Python程序画图:caffe-windows\python\draw_net.py,这个文件,就是用来绘制网络模型的,即将网络模型由prototxt变成一张图片。
在绘制之前,需要先安装两个python库:
(1)安装GraphViz:

conda install -y GraphViz

(2)安装pydot或pydotplus:

conda install -y pydotplus

都安装好了,就可以调用脚本来绘制图片了。

2. 绘制图像:

draw_net.py执行的时候带三个参数:
第1个参数:网络模型的prototxt文件
第2个参数:保存的图片路径及名字
第3个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT,默认为LR 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。

3. 示例:

python F:\Caffe\caffe-2015cuda\python\draw_net.py F:\Caffe\mnist\MyNet_train_test.prototxt F:\Caffe\mnist\MyNet.png --rankdir=TB

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值