caffe网络结构查看神器 caffe网络模型可视化方法

caffe网络结构查看神器 caffe网络模型可视化方法

 方法一：利用Netscope工具

http://ethereon.github.io/netscope/#/editor

把你的.prototxt 文件内容放到左边编辑器中，然后

Shift+Enter刷新，右边的窗口就会显示网络结构图了

 

方法二：利用draw_net.py脚本

前提是安装好caffe，然后在文件夹中可找到draw_net.py脚本，然后将要可视化的网络结构文件放到同一文件夹下，终端中运行以下命令即可显示

python draw_net.py --rankdir TB test.prototxt test.png
其中，TB代表从top画到bottom，test.prototxt是pvanet的网络结构，存储成名字为test.png的图片

rankdir有有四种选项，分别是LR，RL，TB， BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为LR

运行后会生成test.png的图片

第一次运行可能会报和dot相关的错，需要安装以下两个包

pip install pydot
sudo apt-get install graphviz
运行后生成 

参考：https://blog.csdn.net/zong596568821xp/article/details/80996097