caffe入門學習(5):繪製網路結構圖
網路配置都是在prototxt檔案中看起來不直觀,如果能直接看到各層的連線圖就了,Caffe提供了這樣的工具。本文就是描述我用draw_net.py繪製網路結構圖的踏坑過程。
1、安裝graphviz
2)安裝程式
執行安裝程式
完成後將路徑下的 bin 新增到環境變數路徑 path中
2、配置Python模組
1)設定CommonSettings.props
在caffe的配置CommonSettings.props中,開啟python功能支援介面
設定Python的路徑
2)拷貝caffe模組
編譯caffe後,在[caffe_root]\Build\x64\Release
開啟目錄將caffe拷貝到python包管理目錄site-packages中
比如我的Python路徑:
C:\Python27\Lib\site-packages
3)在Python中匯入Caffe模組
開啟python shell執行
import caffe
如果載入失敗,如果提示缺少庫,那麼就用pip2.7進行安裝。
比如:
“ImportError:No module named scipy”
那麼命令列下,進入python目錄,執行 pip2.7 install scipy 就會自動下載安裝缺失的庫
我執行過的操作:
- pip2.7 install numpy
- pip2.7 install scipy
- pip2.7 install matplotlib
- pip2.7 install protobuf
最後import caffe不再報錯就表示配置成功了
3、繪製網路結構圖
1)編寫指令碼
我們這兒還是用 mnis例子來測試,就叫:draw_mnist.bat
python E:/caffe/Build/x64/Release/pycaffe/draw_net.py E:/caffe/caffecase/mnist/lenet_train_test.prototxt E:/caffe/caffecase/mnist/lenetmap .png --rankdir=LRpause
引數說明:
第一個引數:網路定義檔案
第二個引數:生成圖片的儲存名稱和路徑
第三個引數:表示網路繪製方向,有四種,LR 從左到右,RL 從右到左, TB 從上到下 ,BT 從下到上。
2)執行指令碼
雙擊後執行指令碼,結果提示少庫。
ImportError:No module named pydot
提示缺少 pydot模組,同上面方式一樣,用pip進行安裝。
安裝完成後,雙擊後執行又提示報錯。
Exception: “dot.exe” not found in path.
從C:\Graphviz\bin 目錄中,可以發現有這個我們需要的dot.exe,將這個bin目錄新增到系統的環境變數中去即可。
注意:重啟電腦,我之前就是添加了環境變數但是沒重啟,被糾結了我很久。
3)生成結構圖
執行成功,就會生成指定的圖片。