背景
caffe的神經網絡定義是用的protobuf,調理雖然清晰,但是網絡大了以後還是不容易觀看,下面來做一下caffe神經網絡的可視化
自畫圖
前置條件:
- 系統安裝graphviz
-
python安裝pydot
又caffe提供的畫圖工具
/Users/wang/soft/caffe/python/draw_net.py
指令如下:
python draw_net.py ~/Desktop/bak_model/mobilenet_train_test.prototxt ./mobilenet.png --rankdir=BT
- 參數1,神經網絡的protobuf檔案
- 參數2,儲存路徑
- 參數3,輸出格式
- LR,從左到右(left,right)
- RL
- TB,從上到下(to,bottom)
- BT
線上可視化工具
ethereon
http://ethereon.github.io/netscope/quickstart.html
線上工具有一個缺點,輸入資料層容易報錯,這個得忍受一下
ConvNetDraw
https://cbovar.github.io/ConvNetDraw/