drawnet.py繪製網絡結構

腳本在caffe的python目錄裏，須要安裝pydot，graphviz

安裝pydot：

$ sudo apt-get install python-pydot 

安裝graphviz：

 

$ sudo apt-get install graphviz  

 

生成的命令行：

python draw_net.py + 網絡prototxt + 生成圖片名

$ python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

–rankdir選項肯定了輸出圖片的擺放位置，如TB (Top-Bottom) 就是網絡從上到下拜訪，同理RL (Right-Left）就是從右向左，LR (Left-Right) 就是從左向右擺放。默認使用的是LR方式。注意是--。

生成的能夠是jpg，也能夠是png，但生成的文件名必須是絕對路徑，不能直接寫成lenet_structure.jpg，那樣會識別不了。

 

 

繪製模型會報的常見錯誤：

http://blog.csdn.net/yiranyhy/article/details/55509304

 

這裏有一個在線的繪製：

http://ethereon.github.io/netscope/#/editor