在線可視化caffe模型結構

http://ethereon.github.io/netscope/#/editor

假設Caffe的目錄是$(CAFFE_ROOT)

1.編譯caffe的python接口

$ make pycaffe

2.裝各種依賴

$ pip install pydot $ sudo apt-get install graphviz

3.可視化模型

draw_net.py執行的時候帶三個參數

第一個參數：網絡模型的prototxt文件 第二個參數：保存的圖片路徑及名字 第二個參數：--rankdir=x , x 有四種選項，分別是LR, RL, TB, BT 。用來表示網絡的方向，分別是從左到右，從右到左，從上到小，從下到上。默認為ＬＲ。

繪制googlenet模型

$ cd $(CAFFE_ROOT)/python/ $ ./draw_net.py ../models/bvlc_googlenet/deploy.prototxt bvlc_googlenet.caffemodel.jpg

路徑自己改變，./draw_net.py后面接prototxt路徑和可視化圖像的保存路徑。

4.最終的圖像

繪制其他模型

#繪制Lenet模型 cd (CAFFE_ROOT) python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT #繪制cifar10模型 cd $(CAFFE_ROOT) python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

本文轉自：http://blog.csdn.net/ws_20100/article/details/50443566

總結

以上是生活随笔為你收集整理的可视化caffe模型结构及在线可视化的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。