caffe的python編譯模塊提供了繪制net結構的功能，可實現對網絡結構的繪制，這極大的方便了對網絡結構的理解和分析，這里介紹下繪制的過程。

一、準備工作：

1、確保caffe的python模塊已經編譯和配置。

具體可參考【caffe】windows下caffe+vs2013+python2.7+cuda8.0+cmake3.8編譯與配置

2、打開命令行cmd窗口，進入python安裝目錄下的Scripts目錄（這個目錄需要添加到環境變量里，否者pip安裝命令無法使用）。

3、安裝protobuf模塊：pip install protobuf。

4、安裝graphviz模塊：pip install graphviz。

5、安裝graphviz軟件：下載graphviz-2.38.msi安裝文件，安裝，并將安裝目錄下的bin文件夾的路徑添加到環境變量中。

6、安裝pydot模塊和pydotplus模塊：pip install?pydot和pydotplus。

（如果后面的執行依然報錯，建議再安裝pydot-ng模塊）

做好以上準備后，就可以著手進行繪制了，其實這些都是坑，不斷嘗試，最后才跑通，希望后來者能少走點彎路。

二、使用draw_net.py繪制網絡：

1、在自己的工作目錄下新建一個文件夾，命名為draw_net，后面的繪制都將在這個目錄下完成。

2、將caffe安裝目錄中的python文件夾下的draw_net.py文件拷到這個文件夾下。

3、將網絡的模型的txt文件lenet_train_test.prototxt也拷到這個文件夾下。

4、新建一個文本文件，并重命名為draw_net.bat，輸入如下內容：

@echo off draw_net.py lenet_train_test.prototxt lenet.png pause

5、保存，雙擊執行，即可得到如下網絡結構圖。

6、上面得到的是一個橫向排列的網絡結構圖，如果想得到豎向排列的網絡結構圖，可在draw_net.bat輸入如下命令：

@echo off draw_net.py lenet_train_test.prototxt lenet.png --rankdir=BT pause

保存雙擊執行，得到如下網絡結構圖：

三、幾種常見的網絡結構圖
1、alexnet

2、googlenet

3、reference_caffenet

4、ilsvrc13

5、finetune_flickr_style

6、mobilenet

參考：

[1]?https://github.com/fchollet/keras/issues/3210;

[2]?http://www.voidcn.com/blog/tracer9/article/p-4968674.html;

[3]?http://www.itboth.com/d/6nUzmq/python-draw-net.

2017.08.03

總結

以上是生活随笔為你收集整理的【caffe】使用draw_net绘制net结构的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。