绘制网络结构

JLU-IPVR

听笙

安装graphviz

graphviz是贝尔实验室开发的一个开源的绘图工具,它可以很方便绘制结构化的图形网络,支持多种格式输出,如各种常见的图片格式(bmp、png等),PDF,SVG等。 graphviz使用dot作为脚本语言,只需要在dot脚本中定义图的顶点和边,以及形状、颜色、字体、填充等样式,graphviz就可以使用合适的布局算法对图形布局,使各顶点均匀分布在画布上,并且尽可能的减少边的交叉。

下载地址:http://www.graphviz.org/Download_windows.php

下载完成之后解压出来是一个release的文件夹,把release下的bin文件夹的路径E:\caffe-windows\graphviz-2.38\release\bin添加到环境变量里,让系统可以找到graphviz的相关文件就可以了。

注意:要重启电脑。

重启之后可以再次在CMD中执行一遍pip install graphviz

 

安装pydot

pydot模块提供了Python对graphviz dot语言的接口,使在python下可以调用graphviz工具绘图。

pydot下载地址: https://pypi.python.org/pypi/pydot

下载之后解压,在解压出来的pydot-1.2.3文件夹下有一个pydot.py的安装文件,在所在文件夹下运行cmd命令,输入python pydot.py install 完成安装。

 

python draw_net.py[--rankdir] input_netprotofile output_imagefile

其中,--rankdir = x,–rankdir=x, x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。

新建一个draw_net.bat内容:

Python E:\caffe-windows\caffe-master\Build\x64\Release\pycaffe\draw_net.py  --rankdir=BTE:\caffe-windows\caffe-master\examples\mnist\lenet.prototxt lenet.jpg

绘制网络结构_第1张图片


你可能感兴趣的:(caffe学习笔记,win-caffe)