1. Netscope
进入http://ethereon.github.io/netscope/#/editor
将prototxt文件复制到左侧编辑框中,然后shift+enter
2. 终端
首先,安装这些东西:
sudo apt-get install graphviz
sudo pip install pydot
然后,将模型的prototxt文件放到draw_net.py(在caffe里有这个函数,找找)同目录下,终端输入:
python draw_net.py --rankdir TB train.prototxt train.png
如图(部分):