可视化caffe模型结构

假设Caffe的目录是$(CAFFE_ROOT)

1.编译caffe的python接口

$ make pycaffe

2.装各种依赖

$ pip install pydot
$ sudo apt-get install graphviz

3.可视化模型

$ cd $(CAFFE_ROOT)/python/
$ ./draw_net.py ../../models/VGG_CNN_M_1024/test.prototxt VGG_CNN_M_1024.jpg

路径自己改变，./draw_net.py后面接prototxt路径和可视化图像的保存路径。

4.最终的图像。