背景
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
在线工具有一个缺点,输入数据层容易报错,这个得忍受一下