繪制網絡圖通常有兩種方法:
一種是利用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”就能在右邊將網絡結構圖顯示出來
?