本文介绍如何利用Caffe的Python接口结合GraphViz和pydot库来绘制深度学习网络结构图。首先需要安装GraphViz和pydot，然后通过调用draw_net.py脚本并指定网络模型文件、输出图片名称及网络方向参数即可生成清晰的网络结构图。
在确保caffe的Python接口编译正确的情况下
1 安装GraphViz
sudo apt-get install GraphViz
2 安装pydot
sudo pip install pydot
安装好后就可以调用caffe中Python文件夹下的draw_net.py脚本画图了
比如绘制caffe自带的LeNet网络模型
在caffe文件夹下输入：
sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt lenet.png --rankdir=LR
说明：draw_net.py执行的时候带三个三数
第一个参数：网络模型文件
第二个参数：保存的图片的名字
第三个参数：--rankdir=LR或RL/TB/BT 用来表示网络的方向
扫一扫
3052
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
