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