1. 程式人生 > >win7 caffe使用筆記——draw_net.py繪製caffe網路

win7 caffe使用筆記——draw_net.py繪製caffe網路

先提供一個福利,Netscope的線上視覺化工具http://ethereon.github.io/netscope/#/editor僅僅是檢視網路結構的話使用這個工具,去掉#/editor也可以看到alexnet,vgg16,googlenet等網路結構。

照著ubuntu下的步驟測試成功寫下過程:

1.用的是microsoft for caffe版本,pycaffe介面用的Anaconda2,給兩個很好的配置部落格

2.編寫bat檔案,ubuntu下呼叫的是caffe根目錄/python/draw_net.py,而windows下調根目錄/Build/x64/Release/pycaffe/draw_net.py 

執行的三個引數
第一個引數:網路模型的prototxt檔案
第二個引數:儲存的圖片路徑及名字
第三個引數:--rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。

G:
cd G:\caffe-master
SET GLOG_logtostderr=1

python  Build\x64\Release\pycaffe\draw_net.py  .\examples\Planthopper\draw_net.prototxt  examples\Planthopper\net\Planthopper_net.png
pause 

3.新增python安裝目錄到path系統變數

新增python目錄\Scripts到path系統變數

cmd命令 pip install protobuf pydot

bin路徑加到path系統變數

4.再測試成功