ubuntu16.04 绘制caffe net 网络结构

我们在学习深度的网络的时候，为了能有更好的感性认识，可以使用draw_net.py将caffe net 结构绘制出来

安装python所需要的依赖

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

然后

sudo apt-get install graphviz #graphviz是一种很好用的画图工具
sudo su
pip install pydot #如果未安装pip，可以根据提示，自行安装

在安装完成后，我们发现还是不能绘图，那是因为我们使用了anaconda，所有我们需要将/usr/local/lib/python2.7/dist-packages 文件下的内容，全部复制到/home/XXX/anaconda2/lib/python2.7/site-packages（xxx为用户名） 文件下。

测试绘制mnist的LeNet网络层结构命令：

cd caffe/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png --rankdir=TB

效果如下：

常遇问题

此时如果使用

sudo pip install pydot

会提示没有pip这个命令，故希望读者能按博客里的建议安装步骤来，先切换到root用户，然后执行安装命令，如果提示没有pip，就按照提示安装pip就可以了。