在線(xiàn)可視化caffe模型結(jié)構(gòu)

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

假設(shè)Caffe的目錄是$(CAFFE_ROOT)

1.編譯caffe的python接口

$ make pycaffe

2.裝各種依賴(lài)

$ pip install pydot $ sudo apt-get install graphviz

3.可視化模型

draw_net.py執(zhí)行的時(shí)候帶三個(gè)參數(shù)

第一個(gè)參數(shù)：網(wǎng)絡(luò)模型的prototxt文件 第二個(gè)參數(shù)：保存的圖片路徑及名字 第二個(gè)參數(shù)：--rankdir=x , x 有四種選項(xiàng)，分別是LR, RL, TB, BT 。用來(lái)表示網(wǎng)絡(luò)的方向，分別是從左到右，從右到左，從上到小，從下到上。默認(rèn)為ＬＲ。

繪制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

本文轉(zhuǎn)自：http://blog.csdn.net/ws_20100/article/details/50443566

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。