写在前面
在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 可视化更新