caffe框架中自带模型绘制工具:draw_net.py,位于caffe-master/python,此操作在ubuntu 16.04LTS下进行。
- 调用draw_net.py,首先需要安装依赖包(graphviz和pydot),建议按照先后顺序安装。
- 执行draw_net.py,语法: python draw_net.py [/path/to/solver fle] [/path/to/savefile.png]
Note: 可能提示 no module named _caffe,如下图:
Solution: 跳转到caffe根目录,执行命令: make pycaffe,编译后重新执行步骤2