文章作者：Tyan
博客：noahsnail.com ?|? CSDN ?|? 簡書
1. 介紹
Caffe中自帶繪制神經網絡結構圖的工具，主要是在pycaffe中，因此首先要安裝pycaffe，可參考我的另一篇文章，Caffe安裝，除此之外還需要安裝兩個依賴：pydot和graphviz。
graphviz不能通過pip安裝，否則仍然會報下面的錯誤：
Drawing net to lenet.png
Traceback (most recent call last):
File "draw_net.py", line 58, in <module>
main()
File "draw_net.py", line 54, in main
phase)
File "/Users/ltc/Caffe/caffe-master/python/caffe/draw.py", line 244, in draw_net_to_file
fid.write(draw_net(caffe_net, rankdir, ext, phase))
File "/Users/ltc/Caffe/caffe-master/python/caffe/draw.py", line 223, in draw_net
return get_pydot_graph(caffe_net, rankdir, phase=phase).create(format=ext)
File "/Users/ltc/anaconda/lib/python2.7/site-packages/pydot.py", line 1883, in create
prog=prog))
Exception: "dot" not found in path.
2. graphviz安裝
graphviz安裝命令如下（Mac，Linux可用apt-get）：
brew install graphviz
3. pydot安裝
pydot的安裝命令：
pip install pydot
4. 繪制結構圖
繪制神經網絡結構圖：
# 進入Caffe python目錄
$ cd /Users/ltc/Caffe/caffe-master/python
# 繪制結構圖
$ python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png
5. 結果
結果如下：
LeNet網絡結構圖
