caffe將網路模型由protot轉換成變成網路結構模型圖--caffe學習(5)
阿新 • • 發佈:2018-12-31
首先需要安裝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執行