Python/draw_net.py, 這個文件,就是用來繪製網絡模型的。也就是將網絡模型由prototxt變成一張圖片。
在繪製之前,需要先安裝兩個庫
1、安裝GraphViz
# sudo apt-get install GraphViz
注意,這裏用的是apt-get來安裝,而不是pip.
2 、安裝pydot
# sudo pip install pydot
用的是pip來安裝,而不是apt-get
安裝好了,就可以調用腳本來繪製圖片了
draw_net.py執行的時候帶三個參數
第一個參數:網絡模型的prototxt文件
第二個參數:保存的圖片路徑及名字
第二個參數:--rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網絡的方向,分別是從左到右,從右到左,從上到小,從下到上。默認爲LR。
例:繪製Lenet模型
# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT
例:繪製cifar10的模型
# sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT