1. 程式人生 > >caffe入門學習(5):繪製網路結構圖

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

目錄下會生成pycaffe目錄。
這裡寫圖片描述
開啟目錄將caffe拷貝到python包管理目錄site-packages
比如我的Python路徑:

C:\Python27\Lib\site-packages

3)在Python中匯入Caffe模組
開啟python shell執行

import caffe

如果載入失敗,如果提示缺少庫,那麼就用pip2.7進行安裝。
比如:

ImportErrorNo 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)生成結構圖
執行成功,就會生成指定的圖片。
這裡寫圖片描述