背景
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
![](https://img.laitimes.com/img/9ZDMuAjOiMmIsIjOiQnIsICM38FdsYkRGZkRG9lcvx2bjxiNx8VZ6l2cscXUU5UMVdVYxRWbihGZzwEMW1mY1RzRapnTtxkb5ckYplTeMZTTINGMShUYfRHelRHLwEzX39GZhh2css2RkBnVHFmb1clWvB3MaVnRtp1XlBXe0xyayFWbyVGdhd3LcV2Zh1Wa9M3clN2byBXLzN3btg3Pn5GcuATNwETO1EjM0EDOwkTMwIzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
線上可視化工具
ethereon
http://ethereon.github.io/netscope/quickstart.html
線上工具有一個缺點,輸入資料層容易報錯,這個得忍受一下
ConvNetDraw
https://cbovar.github.io/ConvNetDraw/