1. 程式人生 > >使用Pycaffe繪製Caffe網路圖

使用Pycaffe繪製Caffe網路圖

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

import caffe

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

ImportError: No module named _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(這步不可省略)

python setup.py install

輸出:

Finished processing dependenceies for protobuf==3.0.0

設定成功後,我們在python中再試著import caffe,
skimage.io
提示缺少skimage.io模組,說明還需要新增scikit-image。

pip install scikit-image

結果又出錯了,
scikit-image_error

這是什麼鬼?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

Lenet