您的位置:首页 > 理论基础 > 计算机网络

caffe 使能python接口使用draw_net.py绘制网络结构图过程

2016-11-08 21:36 821 查看
1.先去这个网址http://www.graphviz.org/Download_windows.PHP将graphviz-2.38.msi下下来,进行安装,可以发现它的bin目录下有这个我们需要的dot.exe,将这个bin目录添加到系统的环境变量中去即可。

2.最后重启电脑,使环境变量生效。

3.然后安装pydot

pip install pydot


4. 安装GraphViz

pip install GraphViz

5.生成网络结构图

python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

其中,–rankdir选项确定了输出图片的摆放位置,如TB (Top-Bottom) 就是网络从上到下拜访,同理RL (Right-Left)就是从右向左,

LR (Left-Right) 就是从左向右摆放。默认使用的是LR方式。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: