網絡配置都是在prototxt文件中看起來不直觀,如果能直接看到各層的連接圖就了,Caffe提供了這樣的工具。本文就是描述我用draw_net.py繪製網絡結構圖的踏坑過程。
1、安裝graphviz
繪製圖形,首先我們需要一個圖形包。
1)下載安裝包
https://graphviz.gitlab.io/_pages/Download/Download_windows.html
2)安裝程序
執行安裝程序
完成後將路徑下的 bin 添加到環境變量路徑 path中
2、配置Python模塊
1)設置CommonSettings.props
在caffe的配置CommonSettings.props中,打開python功能支持接口
設置Python的路徑
2)拷貝caffe模塊
編譯caffe後,在[caffe_root]\Build\x64\Release 目錄下會生成pycaffe目錄。
打開目錄將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)生成結構圖
運行成功,就會生成指定的圖片。