1. 程式人生 > >caffe之繪製網路模型

caffe之繪製網路模型

在Caffe中,目前有兩種視覺化prototxt格式網路結構的方法:

1. 使用Netscope線上視覺化

2. 使用Caffe提供的draw_net.py

1. 使用Netscope線上視覺化

使用方法:首先開啟這個地址:http://ethereon.github.io/netscope/#/editor,然後把你的描述神經網路結構的prototxt檔案複製到該編輯框裡,最後按shift-enter,這樣就可以直接以圖形方式顯示網路的結構 。

比如,以mnist的LeNet網路結構為例,把Caffe中example/mnist/lenet_train_test.prototxt檔案的內容複製到編譯框,按shift-enter,立即就可以得到視覺化的結構圖。 


這裡寫圖片描述

2. 使用Caffe提供的draw_net.py

python/draw_net.py, 這個檔案,就是用來繪製網路模型的。也就是將網路模型由prototxt變成一張圖片。

1. 安裝graphviz :   sudo apt-get install graphviz

2. 安裝pydot :  sudo pip install pydot

3.執行draw_net.py檔案,帶有三個引數:網路模型的prototxt檔案,儲存圖片的路徑及名字,--rankdir=x,x有四種選擇,分別是LR, RL, TB, BT,用來表示網路的方向,分別是從左到右,從右到左,從上到下,從下到上,預設是LR。

以繪製Lenet模型為例:

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

這裡寫圖片描述

3.總結

使用Netscope繪製出的圖直觀簡潔,易於快速瞭解網路模型,但是缺少層內的細節資訊; 
使用draw_net.py繪製出的結構圖儲存了引數資訊,細節更豐富,但是結構不是很清晰明瞭,這一點在大型模型上的體現尤為明顯