Linux下进行
在caffe/python下会有一个draw_net.py用于绘制prototxt格式的模型结构原理图。
1.安装python的依赖包
打开cmd输入以下内容,若网速不行,容易中断,请多次尝试。
sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
2.安装python依赖的画图工具graphviz
sudo apt-get install graphviz
3.切换到root用户
sudo su
4.安装pydot
pip install pydot
5.将/usr/local/lib/python2.7/dist-packages 文件下的内容,全部复制到home/anaconda2/lib/python2.7/site-packages文件。
如下图所示:
Caffe下绘制模型结构原理图
6.绘制手写数字的LeNet网模型结构图
在draw_net.py所在的路径caffe/python下打开终端输入:
Python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png –rankdir=TB
按回车键可出现模型结构图如下:
Caffe下绘制模型结构原理图

在线呈现
1.直接打开http://ethereon.github.io/netscope/#/editor
2.将prototxt文件复制到左边框里,然后按shift-enter,就可以得到模型结构原理图,如下所示:
Caffe下绘制模型结构原理图

Caffe下绘制模型结构原理图

两种方法比较如下:
法1绘制出的结构图保存了参数信息,细节更丰富,但是结构不是很清晰明了。
法2绘制出的图直观简洁,易于快速了解网络模型,但是缺少层内的细节信息。

相关文章:

  • 2021-04-27
  • 2021-07-23
  • 2021-08-14
  • 2021-07-30
  • 2021-04-28
  • 2022-02-03
  • 2021-12-23
  • 2022-12-23
猜你喜欢
  • 2021-09-20
  • 2021-05-28
  • 2021-05-15
  • 2021-10-19
  • 2021-07-14
  • 2021-05-04
相关资源
相似解决方案