caffe的python編譯模塊提供了繪制net結(jié)構(gòu)的功能，可實(shí)現(xiàn)對網(wǎng)絡(luò)結(jié)構(gòu)的繪制，這極大的方便了對網(wǎng)絡(luò)結(jié)構(gòu)的理解和分析，這里介紹下繪制的過程。

一、準(zhǔn)備工作：

1、確保caffe的python模塊已經(jīng)編譯和配置。

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

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

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

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

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

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

（如果后面的執(zhí)行依然報(bào)錯(cuò)，建議再安裝pydot-ng模塊）

做好以上準(zhǔn)備后，就可以著手進(jìn)行繪制了，其實(shí)這些都是坑，不斷嘗試，最后才跑通，希望后來者能少走點(diǎn)彎路。

二、使用draw_net.py繪制網(wǎng)絡(luò)：

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

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

3、將網(wǎng)絡(luò)的模型的txt文件lenet_train_test.prototxt也拷到這個(gè)文件夾下。

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

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

5、保存，雙擊執(zhí)行，即可得到如下網(wǎng)絡(luò)結(jié)構(gòu)圖。

6、上面得到的是一個(gè)橫向排列的網(wǎng)絡(luò)結(jié)構(gòu)圖，如果想得到豎向排列的網(wǎng)絡(luò)結(jié)構(gòu)圖，可在draw_net.bat輸入如下命令：

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

保存雙擊執(zhí)行，得到如下網(wǎng)絡(luò)結(jié)構(gòu)圖：

三、幾種常見的網(wǎng)絡(luò)結(jié)構(gòu)圖
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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。