1. 程式人生 > >caffe 03 win10 使用 draw_net.py 畫網路圖

caffe 03 win10 使用 draw_net.py 畫網路圖

01 安裝draw_net.py需要的必備元件

caffe提供話網路圖的python工具caffe\python\draw_net.py

draw_net.py依賴 GraphViz pydot 兩個Python元件。並且需要系統有GraphViz執行程式。
在命令列安裝draw_net.py的python依賴元件。(我的系統使用的python3.5.3,如果使用python2.7.XX的使用pip2 install GraphViz Pydot)。

pip3 install GraphViz pydot

為作業系統安裝GraphViz工具。
不安裝 GraphViz 執行程式,執行draw_net.py 工具繪製網路圖時,會提示如下錯誤:

Exception: “dot.exe” not found in path.

C:\Program Files (x86)\Graphviz2.38\bin

C:\Program Files (x86)\Graphviz2.38\bin 加入path環境變數中。

如果是在命令列視窗使用draw_net.py,修改Path環境變數後,要重新開啟命令列視窗,才能使環境變數生效。

02 使用draw_net.py畫網路圖

::cd %caffe_root%
cd D:\git\DeepLearning\caffe
:: 建立存放生成網路圖的目錄
mkdir build\x
64\tmp python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test01.png --rankdir=BT :: draw_net.py 的引數: :: 第一個引數:網路模型的prototxt檔案 :: 第二個引數:儲存的圖片路徑及名字 :: 第三個引數:--rankdir=x,x有四種是LR,RL,TB,BT,表示影象生成方向。分別是從左到右、從右到左、從上到下、從下到上。預設LR。 python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x
64\tmp\lenet_test02.png --rankdir=LR python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test03.png --rankdir=TB python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test04.png --rankdir=BT

生成的網路圖