我的系统环境:Win10+VS2013+Anaconda3(Python=3.5)+Caffe
Caffe环境的搭建和配置,这里就不赘述了。懒人办法:可直接去GitHub下载编译好的Caffe程序。
主要使用Caffe自带的Python程序画图:caffe-windows\python\draw_net.py,这个文件,就是用来绘制网络模型的,即将网络模型由prototxt变成一张图片。
在绘制之前,需要先安装两个python库:
(1)安装GraphViz:
conda install -y GraphViz
(2)安装pydot或pydotplus:
conda install -y pydotplus
都安装好了,就可以调用脚本来绘制图片了。
draw_net.py执行的时候带三个参数:
第1个参数:网络模型的prototxt文件
第2个参数:保存的图片路径及名字
第3个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT,默认为LR 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。
python F:\Caffe\caffe-2015cuda\python\draw_net.py F:\Caffe\mnist\MyNet_train_test.prototxt F:\Caffe\mnist\MyNet.png --rankdir=TB