1. 程式人生 > >caffe將網路模型由protot轉換成變成網路結構模型圖--caffe學習(5)

caffe將網路模型由protot轉換成變成網路結構模型圖--caffe學習(5)

首先需要安裝graphviz,再安裝pydot
在conda命令下:

conda install graphviz
conda install pydot

python/draw_net.py這個檔案,就是caffe官方提供的用來繪製網路模型的。也就是將網路模型由prototxt變成一張圖片。
安裝好了,就可以呼叫指令碼來繪製圖片了
draw_net.py執行的時候帶三個引數

第一個引數:網路模型的prototxt檔案

第二個引數:儲存的圖片路徑及名字

第二個引數:–rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。
通過如下命令繪製自己的網路結構流程圖:

python /caffe/python/draw_net.py train.prototxt train.png --rankdir=BT

可得到網路結構圖如下:
這裡寫圖片描述
繪製經典的網路模型:
1:繪製Lenet模型

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

這裡寫圖片描述
2:繪製cifar10的模型

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

這裡寫圖片描述
這樣可以更加直觀的繪製自己的caffe模型圖,便於理解

3:也可以藉助已有的方法實現框圖視覺化:
3.1:開啟網址 http://ethereon.github.io/netscope/#/editor
3.2.將自己的train_test.prototxt裡的複製貼上到左邊
3.3:然後同時shift+enter執行
這裡寫圖片描述