使用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模組,說明還需要新增scikit-image。
pip install scikit-image
結果又出錯了,
這是什麼鬼?Visual C++ 9.0 is required??怎麼還需要VS2008?機器上裝的VS2013都不行?好吧,還需要安裝一個Micorsoft 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