Caffe下python環境的編譯和使用draw_net.py繪製lenet網絡結構圖

1、安裝python所需的依賴包

    （1）sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose

    （2）再使用pip安裝requirements.txt文件中所指定的包（若之前沒安裝過pip，需要安裝pip：sudo apt-get install python-pip）

            在caffe根目錄下使用命令：

             for req in $(cat python/requirements.txt); do pip install $req; done

           （在這之前需要使用命令su root轉到root用戶，因爲安裝過程需要root權限）

2、Makefile.config文件中，將

     PYTHON_INCLUDE := /usr/include/python2.7 \
　　   /usr/lib/python2.7/dist-packages/numpy/core/include
     改爲
     PYTHON_INCLUDE := /usr/include/python2.7 \
　　   /usr/local/lib/python2.7/dist-packages/numpy/core/include

3、進行編譯

      make pycaffe

      （如果編譯過程出錯，則可能是依賴包沒裝好，我之前出現過protobuf找不到的錯誤，此時用pip install protobuf安裝重新安裝一下就好）

4、驗證

      進入python文件夾：cd python

      使用命令：python 

                        import caffe

      （如果沒有出錯，則說明安裝好了）

5、使用draw_net.py繪製lenet網絡結構圖

      命令：python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png

      此時出現錯誤：ImportError: No module named pydot，則再次使用pip安裝，pip install pydot

      再次運行：python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png

      又出現錯誤了：pydot.InvocationException: GraphViz's executables not found，坑爹。此時需要安裝graphviz，命令爲sudo apt-get install graphviz

      最後再運行一次，終於成功了。在python文件夾下得到lenet.png文件了。如下圖：