caffe网络结构图绘制
生活随笔
收集整理的這篇文章主要介紹了
caffe网络结构图绘制
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
繪制網絡圖通常有兩種方法:
一種是利用python自帶的draw_net.py,首先安裝兩個庫:
sudo apt-get install graphviz
sudo pip install pydot
接下來就可以用python自帶的draw_net.py文件來繪制網絡圖了。
draw_net.py執(zhí)行時帶三個參數:
一、網絡模型的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”就能在右邊將網絡結構圖顯示出來
?
總結
以上是生活随笔為你收集整理的caffe网络结构图绘制的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: GDB调试技术(一)
- 下一篇: (十八)深入浅出TCPIP之epoll的