用Caffe自带程序画网络结构图

我的系统环境:Win10+VS2013+Anaconda3(Python=3.5)+Caffe
Caffe环境的搭建和配置,这里就不赘述了。懒人办法:可直接去GitHub下载编译好的Caffe程序。

1. 画图工具:

主要使用Caffe自带的Python程序画图:caffe-windows\python\draw_net.py,这个文件,就是用来绘制网络模型的,即将网络模型由prototxt变成一张图片。
在绘制之前,需要先安装两个python库:
(1)安装GraphViz:

conda install -y GraphViz

(2)安装pydot或pydotplus:

conda install -y pydotplus

都安装好了,就可以调用脚本来绘制图片了。

2. 绘制图像:

draw_net.py执行的时候带三个参数:
第1个参数:网络模型的prototxt文件
第2个参数:保存的图片路径及名字
第3个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT,默认为LR 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。

3. 示例:

python F:\Caffe\caffe-2015cuda\python\draw_net.py F:\Caffe\mnist\MyNet_train_test.prototxt F:\Caffe\mnist\MyNet.png --rankdir=TB

这里写图片描述

猜你喜欢

转载自blog.csdn.net/cosmoshua/article/details/72829954