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

原创 2017年01月02日 23:19:58

首先需要安装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运行
这里写图片描述

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

浅谈caffe中train_val.prototxt和deploy.prototxt文件的区别

本文以CaffeNet为例: 在刚开始学习的时候,觉得train_val.prototxt文件和deploy.prototxt文件很相似,然后当时想尝试利用deploy.prototxt还原出trai...

迁移学习技巧以及如何更好的finetune 模型

最近在finetune model的时候遇到了点问题,开贴记录一下。也算填自己踩过的坑。 文章参考翻译自cs231n 其实我们常用的直接finetune pre-trained model就属于迁...

caffe/tensorflow中样本的label一定要从序号0开始标注--caffe学习(15)

这两天在跑实验时思考一个问题,为什么在别的帖子里面和自己之前的实验中,对于data的标注大家都默认使用的是从0开始标注样本,之前的一次finetune中,自己的样本从34567开始标注的时候一直没有开...

caffe将用训练好的caffemodel和train_val.prototxt文件分类新的一张图片-上篇--caffe学习(6)

经过前文介绍之后可以训练得到caffemodel文件,同时得到train_val.prototxt文件,但是我们分类时需要的是deploy.prototxt文件。 因此需要手动将train_val....

caffe中train_val.prototxt文件和deploy.prototxt文件区别和转换--caffe学习(14)

先放出二者的完整例子文件,然后分析: train_val.prototxt文件如下:name: "CaffeNet" layer { name: "data" type: "Data" top: "...

如何进行finetune

进行finetune的命令如下: ..\..\bin\caffe.exe train --solver=.\solver.prototxt -weights .\test.caffemodel p...

caffe中loss函数代码分析--caffe学习(16)

http://blog.csdn.net/u014381600/article/details/54341317接上篇:caffe中样本的label一定要从序号0开始标注–caffe学习(15) A...

Caffe-使用C++的分类接口

在训练完一个模型之后,如果想利用现有的模型文件、网络参数文件去分类测试一张图片,caffe给出了C++的接口,其c++代码在 examples/cpp_classification/clssifica...

caffe的solver文件参数详解--caffe学习(2)

solver文件参数:test_interval,test_iter,max_iter的区别: test_interval = 938 #60000/64,测试间隔参数:训练完一次所有的图片,进...

Python读取MATLAB存储的mat文件

最近项目中需要在python下读取mat文件,遇到以下问题,现在已经成功使用python的h5py库完全解决: 首先,尝试了文章1中的scipy.io库来读取mat文件,但是出现了: 1:Plea...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)