Caffe学习系列(18): 绘制网络模型
2016-12-29 00:02
211 查看
python/draw_net.py, 这个文件,就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。
在绘制之前,需要先安装两个库
1、安装GraphViz
注意,这里用的是apt-get来安装,而不是pip.
2 、安装pydot
用的是pip来安装,而不是apt-get
安装好了,就可以调用脚本来绘制图片了
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
例:绘制Lenet模型
例:绘制cifar10的模型
在绘制之前,需要先安装两个库
1、安装GraphViz
# sudo apt-get install GraphViz
注意,这里用的是apt-get来安装,而不是pip.
2 、安装pydot
# sudo pip install pydot
用的是pip来安装,而不是apt-get
安装好了,就可以调用脚本来绘制图片了
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:--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
相关文章推荐
- Caffe学习系列(18): 绘制网络模型
- Caffe学习系列(18): 绘制网络模型
- Caffe学习系列(18): 绘制网络模型
- Caffe学习系列(1): 绘制网络模型
- Caffe学习系列 绘制网络模型
- Caffe学习系列 绘制网络模型
- Caffe(18)——绘制网络模型
- Caffe绘制网络模型ResNet_18_deploy
- Caffe学习系列——工具篇:神经网络模型结构可视化
- Caffe绘制网络模型ResNet_18_train_val
- caffe学习系列:绘制网络模型
- Andrew NG机器学习课程笔记系列之——机器学习之神经网络模型-上(Neural Networks: Representation)
- 技术系列之 网络模型(二)
- 绘制网络模型流程图
- 【原创】技术系列之 网络模型(二)
- 【计算机网络系列(三)】TCP/IP 层次模型和相关协议
- webkit源码分析系列-css盒模型和元素绘制
- Andrew NG机器学习课程笔记系列之——机器学习之神经网络模型-下(Neural Networks: Representation)
- 【原创】技术系列之 网络模型(三)多路复用模型
- webkit源码分析系列-css盒模型和元素绘制