caffe网络可视化的多种方式

文章目录

    • 写在前面
    • draw_net pycaffe工具
    • Netscope 在线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 可视化更新

你可能感兴趣的:(机器学习,caffe学习手记)