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



©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页