caffe入門學習(5):繪製網絡結構圖

網絡配置都是在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進行安裝。
比如:

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

發佈了155 篇原創文章 · 獲贊 59 · 訪問量 22萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章