1. 程式人生 > >caffe 使能python介面使用draw_net.py繪製網路結構圖過程

caffe 使能python介面使用draw_net.py繪製網路結構圖過程

前一章很不容易把python介面配置好了,接下來做測試,繪製網路結構圖。又遇到各種奇葩問題。

繪製命令如下:

python draw_net.py deploy.prototxt net.bmp --rankdir=BT

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

記錄如下:

1.cmd到....\caffe-master\Build\x64\Release\pycaffe

目錄下,可以看到有我們需要的draw_net.py,然後可以將models\bvlc_googlenet\train_val.prototxt拷貝到這個目錄下,輸入上述命令,會報錯Import Error:No module named protobuf.,然後看到外國友人也問了這一帖子,網址連結http://stackoverflow.com/questions/38680593/importerror-no-module-named-google-protobuf。出現錯誤的原因When pip tells you that you already have protobuf, but PyCharm (or other) tells you that you don't have it, it means that 
pip and PyCharm are using a different Python interpreter. 參照這個人講的


他是在linux下編的,然後百度了一下windows virtualenv.找到這個http://www.2cto.com/os/201411/355798.html,就是在windows配置python的虛擬機器環境。

2.按照上面方式配置好windows下的python環境後,接下來還是按照http://blog.csdn.net/lg1259156776/article/details/52563889這個網址提供的在虛擬機器環境下,把那些包重新下載一遍,下載完成後重複第一步一開始的操作,這時會報dot.exe not found in path,這時又去解決這個錯誤,看這個網址的http://stackoverflow.com/questions/13987353/python-valueerror-program-dot-not-found-in-path

先去這個網址http://www.graphviz.org/Download_windows.php將graphviz-2.38.msi下下來,進行安裝,可以發現它的bin目錄下有這個我們需要的dot.exe,將這個bin目錄新增到系統的環境變數中去即可,再pip install  pygraphviz即可。

3.最後重啟電腦,使環境變數生效,再次執行即可,得到我們想要看到的網格結構圖。


大功告成,網格圖還是很he人的哈~~~