法一:
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 是绘制后模型的图片
法三:
Netron 可视化工具,直接把模型(prototxt 或者 caffemodel)拖拽进工作区即可
https://github.com/lutzroeder/netron
对 prototxt 和 caffemodel 的操作可以参考
caffe:生成、修改prototxt及caffemodel解析完整版
参考
Ubutun 下 Anaconda3 安装 graphviz 以便 jupyter notebook使用
Caffe—自带工具进行网络结构(xxx.prototxt)可视化