Caffe 的可视化 （二）网络结构可视化

Caffe 的可视化 （二）网络结构可视化

使用Caffe里的工具 draw_net.py   (该文件在目录 $(CAFFE_ROOT)/python/ 里)

usage: draw_net.py [-h] [--rankdir RANKDIR] [--phase PHASE]
                   input_net_proto_file  output_image_file
以LeNet为例：

mark@ubuntu:~/caffe$ python  ./python/draw_net.py    ./examples/mnist/lenet_train_test.prototxt    mnist.png

生成的网络结构图 mnist.png 如下：

也可以加上参数 rankdir

mark@ubuntu:~/caffe$ python   ./python/draw_net.py   --rankdir   BT   ./examples/mnist/lenet_train_test.prototxt   mnist2.png

生成的网络结构图 mnist2.png 如下：

还可以使用下面的网站来查看网络结构图

http://ethereon.github.io/netscope/#/editor