1.環境準備: 編譯好了pycaffe 2.繪製網路結構圖: 終端輸入: cd caffe /python

~/caffe/python$ python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt caffenet.png

Drawing net to caffenet.png

`~/caffe/python$ python draw_net.py ../models/bvlc_alexnet/train_val.prototxt alexnet.png

Drawing net to alexnet.png

python draw_net.py ../models/bvlc_googlenet/train_val.prototxt googlenet.png
 

Drawing net to googlenet.png

python draw_net.py ../models/bvlc_reference_rcnn_ilsvrc13/deploy.prototxt rcnn.png

Drawing net to rcnn.png

python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet5.png

Drawing net to lenet5.png

python draw_net.py ../examples/mnist/mnist_autoencoder.prototxt mnist_ae.png
 

Drawing net to mnist_ae.png

python draw_net.py ../examples/cifar10/cifar10_full_sigmoid_train_test_bn.prototxt cifar10_full_sigmoid_bn.png

Drawing net to cifar10_full_sigmoid_bn.png

3.回到 caffe/python 目錄下就可以看到了 xx.png了