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文件了。如下图：