1. 程式人生 > >caffe網路結構圖繪製

caffe網路結構圖繪製

繪製網路圖通常有兩種方法:

一種是利用python自帶的draw_net.py,首先安裝兩個庫:

sudo apt-get install graphviz

sudo pip install pydot

接下來就可以用python自帶的draw_net.py檔案來繪製網路圖了。

draw_net.py執行時帶三個引數:

一、網路模型的prototxt檔案;

二、儲存的圖片路徑和名字;

三、--rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。

在終於輸入命令:sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt test.png --rankdir=TB 另一種是利用Netscope繪製: 直接登入網站:http://ethereon.github.io/netscope/#/editor

在左邊編輯框裡面將prototxt的資訊貼上進來,然後按“shift+enter”就能在右邊將網路結構圖顯示出來