Caffe net結構可視化

簡介

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

結果
LeNetsvg

本文同步分享在 博客「無比機智的永哥」(CSDN)。
若有侵權,請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」,歡迎正在閱讀的你也加入,一塊兒分享。工具

相關文章
相關標籤/搜索