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 
 

猜你喜欢

转载自blog.csdn.net/qq_41895190/article/details/88855969