python繪製caffe例項的網路模型
阿新 • • 發佈:2019-01-08
在所有目錄與檔案均是基於caffe根目錄!!!!!
python/draw_net.py可以將網路模型由prototxt變成一張圖片。
在繪製之前,先安裝兩個庫:::
1、安裝GraphViz
sudo apt-get install GraphViz
Graphviz的是一款圖形繪製工具,用來被python程式呼叫繪製圖片。
2、安裝pydot
sudo pip install pydot
pydot是python的支援畫圖的庫
編譯pycaffe
在繪製之前,還要執行下述語句編譯_caffe.cpp
make pycaffe
否則會報如下圖所示錯誤:::
完成上述三步後,就可以呼叫指令碼來繪製圖片了
draw_net.py執行的時候帶三個引數
第一個引數:網路模型的prototxt檔案
第二個引數:儲存的圖片路徑及名字
第二個引數:–rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。
繪製Lenet模型
python python/draw_net.py examples/mnist/lenet_train_test.prototxt ./sc.jpg --rankdir=BT
繪製cifar10的模型
python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt cifar10_net.jpg --rankdir=BT