python繪制caffe中網絡模型

caffe-master/python/draw_net.py 實作繪制caffe中定義的網絡模型功能，将.prototxt檔案可視化。

需要先安裝pydot和protobuf工具

通過Anaconda安裝pydot和protobuf工具：

sduo chmod 777 -R ~/anaconda2
conda install protobuf
conda install pydot           

如果還缺少其他工具，也可以通過這種方式安裝。

繪制Lenet網絡模型

MNIST手寫體網絡模型Lenet：

cd ~/caffe-master
./python/draw_net.py ./examples/mnist/lenet_train_test.prototxt ./python/lenet.png           

cifar：

cd ~/caffe-master
./python/draw_net.py ./examples/cifar10/cifar10_quick_train_test.prototxt ./python/cifar.png           

更改網絡模型方向顯示

• 第一個參數：網絡模型的prototxt檔案

• 第二個參數：儲存的圖檔路徑及名字

• 第二個參數：--rankdir=x , x 有四種選項，分别是LR, RL, TB, BT 。用來表示網絡的方向，分别是從左到右，從右到左，從上到小，從下到上。預設為LR。

更改繪制方向為從下到上：

Lenet：