caffe之繪製網路模型
阿新 • • 發佈:2019-01-23
在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繪製出的結構圖儲存了引數資訊,細節更豐富,但是結構不是很清晰明瞭,這一點在大型模型上的體現尤為明顯