天天看點

使用Pycaffe繪制Caffe網絡圖

檢驗Caffe能否在Python中使用,我們可以在Python中試着import caffe。

import caffe
           

如果成功的話就可以直接跳到文末看畫圖就好了,失敗的話往往會顯示:

則說明caffe路徑未設定正确,那麼就一起來配置一下吧。我們需要先設定系統環境變量PYTHONPATH,是你的caffe目錄下的python檔案夾 $(caffe)\python。再将$(caffe)\Build\x64\Release\pycaffe\caffe 目錄生成的所有檔案拷貝到$(caffe)\python\caffe目錄中。

我們需要下載下傳安裝google.protobuf,位址在google protobuf releases。下載下傳其中的protobuf-python-3.1.0.zip和protoc-3.1.0-win32.zip,各自解壓縮,并将win32包中的protoc.exe放在python包的setup.py同一目錄下,再執行build指令。

python setup.py build
           

執行test,确認編譯正确

python setup.py test
           

輸出:

----------------------------------------------------------------------
Ran 590 tests in 1.079s

OK (skipped=7)
           

繼續執行install(這步不可省略)

輸出:

Finished processing dependenceies for protobuf==.
           

設定成功後,我們在python中再試着import caffe,

使用Pycaffe繪制Caffe網絡圖

提示缺少skimage.io子產品,說明還需要添加scikit-image。

pip install scikit-image
           

結果又出錯了,

使用Pycaffe繪制Caffe網絡圖

這是什麼鬼?Visual C++ 9.0 is required??怎麼還需要VS2008?機器上裝的VS2013都不行?好吧,還需要安裝一個Micorsoft Visual C++ Compiler for Python 2.7的包,去微軟官網下載下傳一個吧

Microsoft Visual C++ Compiler for Python 2.7

結果你會發現,還需要安裝lapack/blas等等各種包,簡直是要抓狂。這裡給大家推薦一個Python擴充庫站點,

Unofficial Windows Binaries for Python Extension Packages

建議下載下傳對應的whl,先安裝Numpy + mkl,然後再分别安裝Scipy、scikit-image、pydot、Graphviz。Graphviz需要安裝一個msi包,下載下傳位址為 Graph Visualization Software。其中如果提示找不到dot.exe,那麼還需要将Graphviz2.38\bin添加到系統環境變量PATH中。這一切都安裝完成後再去

import caffe
           

終于成功了!忙了大半天是不是都忘了我們今天是要做什麼了。。。

不是說好要畫圖的嘛。。。

執行繪制網絡圖指令:

draw_net.py ..\examples\mnist\lenet_train_test.prototxt lenet.png
           
使用Pycaffe繪制Caffe網絡圖