caffe中网络结构的可视化

来源:互联网 发布:messenger mac 编辑:程序博客网 时间:2024/06/05 20:06

实验工具是 ubuntu系统下的caffe

可视化方法很多,一种是用caffe自带的draw_net.py 来实现网络结构的可视化。具体实现如下

>> python  /caffe/python/draw_net.py  train.prototxt  net.png  
用python命令执行draw_net.py   实现train.prototxt 的网络结构 并保存在net.png 中,net.png 在caffe/python的文件夹下。这样画出来的是从左到右 代码后加上-rankdir BT可以实现从下到上,看你想怎么看吧



还有一种是GitHub上ethereon开发的一个神经网络可视化的好工具Netscope

(网址:http://ethereon.github.io/netscope/#/editor),只需要把prototxt文件内容复制到左边的框框中,然后同时按住shift+enter即可在右边生成网络的结构图。


在netscope的基础上,GitHub上的dgschwend又开发了可视化+分析网络参数、中间变量等功能的加强版(网址:https://dgschwend.github.io/netsope/quickstart.html)