Chapter 7 Windows下pycaffe的使用之draw_net.py


Chapter 6 中完成了在Windows下,對pycaffe的編譯,如果編譯存在問題,請參考:http://www.cnblogs.com/xiaopanlyu/p/6158902.html

本文接下來的內容假設已經編譯好了pycaffe,直接進行使用。

1. 環境配置

為了和chapter 6 中編譯pycaffe時使用的Python版本保持一致,這里使用的Python版本是 Python2.7.12-64bit.下載地址:https://www.python.org/downloads/release/python-2712/

1) 安裝pip

在Windows cmd命令行中輸入以下命令,本文以下命令未特別說明,均在此命令行下執行

命令:python -m pip install -U pip

2) 安裝protobuf

命令:pip install protobuf

如果 版本不夠高的話,可能還是會出現importerror的錯誤,可以使用升級

命令:pip install protobuf --upgrade

3) 安裝pydot

命令:pip install pydot

4) 安裝GraphViz

命令:pip install GraphViz

5) 安裝scikit_image

命令:pip install scikit-image

如果安裝出錯,可以下載后本地安裝,下載地址:http://www.lfd.uci.edu/~gohlke/pythonlibs

從這個網站下載scikit_image-0.12.3-cp27-cp27m-win_amd64.whl(對應python x64),此處建議也直接把6) 、7) 、8) 中葯安裝的包下載下來,直接本地安裝

下載之后進行本地安裝,注意文件路徑

命令:pip install D:\your path\scikit_image-0.12.3-cp27-cp27m-win_amd64.whl

6) 安裝scipy

命令:pip install D:\your path\scipy-0.18.0-cp27-cp27m-win_amd64.whl

7) 安裝numpy-1.11.1+mkl

命令:pip install D:\your path\numpy-1.11.1+mkl-cp27-cp27m-win_amd64.whl

8) 安裝matplotlib

命令:pip install D:\your path\matplotlib-1.5.3‑cp27‑cp27m‑win_amd64.whl

2. Draw Net

將network.prototxt復制到Release文件夾下,cmd到該路徑下,根據以下使用提示繪制網絡結構。

usage: draw_net.py [-h] [--rankdir RANKDIR] [--phase PHASE]

                   input_net_proto_file output_image_file

1) draw LeNet-5

命令:python draw_net.py --rankdir=BT lenet_train_test.prototxt LeNet-5.bmp

LeNet-5

2) draw CIFAR10_full

命令:python draw_net.py --rankdir=BT cifar10_full_train_test.prototxt cifar10_full.bmp

cifar10_full

3) draw CaffeNetConv

命令:python draw_net.py --rankdir=BT bvlc_caffenet_full_conv.prototxt bvlc_caffenet.bmp

bvlc_caffenet

3. 他山之石

如果覺得使用draw_net.py比較麻煩,推薦一個在線可視化的網站:http://ethereon.github.io/netscope/quickstart.html

當然各有優缺點,自行選擇。

Reference

1. http://blog.csdn.net/LG1259156776/article/details/52563889

2. http://www.lfd.uci.edu/~gohlke/pythonlibs/

3. https://www.python.org/downloads/release/python-2712/

4. http://www.cnblogs.com/xiaopanlyu/p/6158902.html


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM