caffe网络可视化的多种方式

版权声明:本文为博主 一銤阳光 学习整理的文章,如需转载,请注明出处、附上CSDN博文链接。 https://blog.csdn.net/CSDNhuaong/article/details/83689622

写在前面

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

猜你喜欢

转载自blog.csdn.net/CSDNhuaong/article/details/83689622