Caffe 网络结构可视化

法一:

netscop 在线工具,网址如下

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

复制好 model.prototxt 的内容,Shift+Enter 就可以画出网络结构

法二:

caffe 自带的画图工具,在 /caffe/python/ 目录下,draw_net.py 文件,使用该方法需要提前安装两个库

apt-get install graphviz
pip install pydot

绘制方法如下

运行 draw_net.py

python  caffe/python/draw_net.py model.prototxt model.png

model.prototxt 是想要绘制的模型,model.png 是绘制后模型的图片

参考

Ubutun 下 Anaconda3 安装 graphviz 以便 jupyter notebook使用
Caffe—自带工具进行网络结构(xxx.prototxt)可视化

你可能感兴趣的:(Caffe 网络结构可视化)