用caffe/python/draw_net.py可以画caffe的网络结构,
不过好事多磨,提示no module pydot
我是安装的anaconda,所以这样安装
conda install -c https://conda.binstar.org/sstromberg pydot
又提示没有graphviz,那就装 sudo apt-get install graphviz
wzy@wzy-ubuntu:~/caffe/python$ ./draw_net.py ../examples/mnist/mnist_autoencoder.prototxt mnist.png
Drawing net to mnist.png
wzy@wzy-ubuntu:~/caffe/python$ ./draw_net.py ../examples/mnist/lenet.prototxt mnist-lenet.png
wzy@wzy-ubuntu:~/caffe/python$ ./draw_net.py ../models/bvlc_reference_caffenet/deploy.prototxt c.png