簡介
net可視化主要是以流程圖的形式把caffe中使用的模型(每一個layer以及layer和layer之間的鏈接)展示出來,這樣更加直觀和具體。caffe的python接口下面已經提供了可視net 的功能,就是’draw_net.py’文件。python
編譯pycaffe
編譯caffe時必須編譯了caffe的python的wrapper,命令以下,這樣才能正常使用caffe下的python接口web
make pycaffe
安裝pydot
python的pydot模塊是Graphviz的dot語言的python接口,這裏主要使用Graphviz繪圖,固然要安裝pydot
網址:https://pypi.python.org/pypi/pydot
使用下面命令安裝。bash
source ENV/bin/activate
pip install pydot
安裝Graphviz
Graphviz是大名鼎鼎的貝爾實驗室的幾位牛人開發的一個畫圖工具。它的理念和通常的「所見即所得」的畫圖工具不同,是「所想即所得」。Graphviz提供了dot語言來編寫繪圖腳本。
網址:http://www.graphviz.org/
能夠根據本身的系統下載相應的deb文件安裝。
下載源碼根據INSTALL編譯安裝,執行下面命令進行編譯安裝。網絡
cd graphviz_dir
./configure
make
make check
sudo make insall
make installcheck
繪製
繪製mnist的LeNet網絡層結構app
cd caffe_root/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png
結果
svg
本文同步分享在 博客「無比機智的永哥」(CSDN)。
若有侵權,請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」,歡迎正在閱讀的你也加入,一塊兒分享。工具