caffe網路視覺化的多種方式
阿新 • • 發佈:2018-11-27
文章目錄
寫在前面
在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 視覺化更新