caffe系列(如何画网络结构图)
1、安装GraphViz
- mac :
brew install graphviz
- linux:
sudo apt-get install GraphViz
2、安装 pydot
pip install pydot
3、绘制Lenet模型
sudo python /Users/douglaswang/github/caffe-master/python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。