轉(zhuǎn)載自：

Caffe學(xué)習(xí)系列(18): 繪制網(wǎng)絡(luò)模型 - denny402 - 博客園
http://www.cnblogs.com/denny402/p/5106764.html

python/draw_net.py, 這個(gè)文件，就是用來繪制網(wǎng)絡(luò)模型的。也就是將網(wǎng)絡(luò)模型由prototxt變成一張圖片。

在繪制之前，需要先安裝兩個(gè)庫

１、安裝ＧraphViz

# sudo apt-get install GraphViz

注意，這里用的是apt-get來安裝，而不是pip.

2 、安裝pydot

# sudo pip install pydot

用的是pip來安裝，而不是apt-get

安裝好了，就可以調(diào)用腳本來繪制圖片了

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 。用來表示網(wǎng)絡(luò)的方向，分別是從左到右，從右到左，從上到小，從下到上。默認(rèn)為ＬＲ。

例：繪制Lenet模型

# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

例：繪制cifar10的模型

# sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

總結(jié)

以上是生活随笔為你收集整理的Caffe学习系列(18): 绘制网络模型的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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