使用caffe编写的深度网络可视化方法

使用caffe编写的深度网络可视化方法

方法1:Netscope(优点:在线查看可视化网络结构)

Netscope是个支持prototxt格式描述的神经网络结构的在线可视工具,直接打开链接 http://ethereon.github.io/netscope/#/editor 如图,在红色框内复制描述神经网络结构的.prototxt文件,按下“shift+enter”键,即可生成
使用caffe编写的深度网络可视化方法
使用caffe编写的深度网络可视化方法

方法2 安装插件(优点:可以保存可视化网络结构)

ubuntu下安装

sudo apt-get install graphviz
sudo pip install pydot

cd 命令

python /自己的caffe工程/python/draw_net.py /自己的caffe工程下的深度网络/train.prototxt /home/net.png --rankdir=(LR, RL, TB, BT)

其中 “–rankdir=(LR, RL, TB, BT)”用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
参考:
1、https://blog.****.net/10km/article/details/52713027
2、https://blog.****.net/langb2014/article/details/50970520