个人技术笔记

画出caffe的模型的网络架构图

发布时间：2016-12-15  栏目：Caffe, 深度学习  评论：0 Comments

在caffe中可以使用draw_net.py轻松地绘制卷积神经网络（CNN，Convolutional Neural Networks）的架构图。这个工具对于我们理解、学习甚至查错都有很大的帮助。

draw_net.py的使用方法如下：

usage: draw_net.py [-h] [--rankdir RANKDIR]
                   input_net_proto_file output_image_file

Draw a graph of the net architecture.

positional arguments:
  input_net_proto_file  Input network prototxt file
  output_image_file     Output image file

optional arguments:
  -h, --help            show this help message and exit
  --rankdir RANKDIR     One of TB (top-bottom, i.e., vertical), RL (right-
                        left, i.e., horizontal), or another valid dot option;
                        see
                        http://www.graphviz.org/doc/info/attrs.html#k:rankdir
                        (default: LR)

其中，–rankdir选项确定了输出图片的摆放位置，如TB (Top-Bottom) 就是网络从上到下拜访，同理RL (Right-Left）就是从右向左，LR (Left-Right) 就是从左向右摆放。默认使用的是LR方式。

 

画出caffe的模型的网络架构图的步骤：

1. 按照官网的安装教程装好caffe的python库，在caffe的目录下sudo make pycaffe
2. sudo apt-get install python-pydot
3. $ cd caffe/python/
   $ python draw_net.py –rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

留下评论

You must be logged in to post a comment.