1. 程式人生 > >caffe網路視覺化的多種方式

caffe網路視覺化的多種方式

文章目錄

寫在前面

在caffe網路的訓練和除錯的過程中,將網路結構視覺化,可以更加直觀地進行網路調整,將caffe網路結構視覺化也有多種方式,下面來一一介紹。

draw_net pycaffe工具

caffe提供了基於Python的網路結構視覺化工具,只需要編譯pycaffe後,便可以直接使用。
工具位置: {CAFFE_ROOT}/python

使用方法

python    draw_net.py    --rankdir   TB  \
 /home/hualong/net/srelu-resnet-18/srelu-resnet-18.prototxt   \
 srelu-resnet-18.png

Netscope 線上caffe網路視覺化編輯工具

如果說draw_net工具可以將caffe網路結構以圖片形式可視化出來,那麼Netscope更強大的地方,不僅可以將網路結構可視化出來,還能線上實時修改編輯修改,再視覺化

工具連結: http://ethereon.github.io/netscope/quickstart.html

使用方式

  • 載入caffe網路模型
  • shift + enter 視覺化更新