参考http://blog.csdn.net/happynear/article/details/45440709

1. 安装pydot:

sudo pip install pydot

 2. 安装Graphviz:

sudo apt-get install GraphViz

 3.调用并显示

sudo python draw_net.py ../examples/mnist/lenet_train_test.prototxt 1.jpg

 显示结果

使用GraphViz画caffe网络结构图

相关文章: