关闭

caffe将网络模型由protot转换成变成网络结构模型图--caffe学习(5)

标签: 图片网络caffe
482人阅读 评论(0) 收藏 举报
分类:

首先需要安装graphviz,再安装pydot
在conda命令下:

conda install graphviz
conda install pydot

python/draw_net.py这个文件,就是caffe官方提供的用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。
安装好了,就可以调用脚本来绘制图片了
draw_net.py执行的时候带三个参数

第一个参数:网络模型的prototxt文件

第二个参数:保存的图片路径及名字

第二个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
通过如下命令绘制自己的网络结构流程图:

python /caffe/python/draw_net.py train.prototxt train.png --rankdir=BT

可得到网络结构图如下:
这里写图片描述
绘制经典的网络模型:
1:绘制Lenet模型

python /caffe/python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

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

python /caffe/python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

这里写图片描述
这样可以更加直观的绘制自己的caffe模型图,便于理解

3:也可以借助已有的方法实现框图可视化:
3.1:打开网址 http://ethereon.github.io/netscope/#/editor
3.2.将自己的train_test.prototxt里的复制粘贴到左边
3.3:然后同时shift+enter运行
这里写图片描述

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:44933次
    • 积分:155
    • 等级:
    • 排名:千里之外
    • 原创:45篇
    • 转载:11篇
    • 译文:0篇
    • 评论:13条
    最新评论